当前位置：网站首页>pytorch查看张量占用内存大小

pytorch查看张量占用内存大小

element_size返回单个元素的字节大小，nelement返回元素个数

import torch

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

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

版权声明
本文为[保住头发的破风小子]所创，转载请带上原文链接，感谢
https://blog.csdn.net/weixin_43486780/article/details/125570248