Pytorch view tensor memory size

element_size Returns the byte size of a single element ,nelement Number of return elements

import torch

a = torch.zeros([128, 128])
print(a.element_size() * a.nelement())

a = torch.zeros([256, 256])
print(a.element_size() * a.nelement())