获取在 PyTorch 中存储张量所需的内存

问题描述 投票:0回答:0
import torch
import sys

a = torch.rand(10)

sys.getsizeof(a) # 72
sys.getsizeof(a.storage()) # 88
a.element_size() * a.nelement() # 40

谷歌搜索后,我发现人们使用这 3 种方法中的一种,但他们都返回 3 种不同的输出。

哪个是正确的?

memory pytorch tensor
© www.soinside.com 2019 - 2024. All rights reserved.