khakhlyuk/pytorch_tensor_size.py

Created November 17, 2019 20:44

Star (0) You must be signed in to star a gist
Fork (0) You must be signed in to fork a gist

Select an option

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/khakhlyuk/9fef8630883e26d1adba15c895540d5f.js"></script>
Save khakhlyuk/9fef8630883e26d1adba15c895540d5f to your computer and use it in GitHub Desktop.

Download ZIP

Memory size of a pytorch tensor

Raw

pytorch_tensor_size.py

x.element_size() * x.nelement()

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment