在学pytorch的过程中遇到了要求tensor元素个数的操作,从而有了以下代码:
import torch
import numpy as np
def test():
torch.manual_seed(1)
a = torch.randn(3, 4)
print(a)
print(a.size())
print(*a.size())
print(np.multiply(*a.size()))
test()
输出:
所以* 表示是解包,将size的[3,4]解包为3,4。之后再进行相乘操作,打包解包详解参见https://www.cnblogs.com/mo-nian/p/11842422.html
好了,以后求tensor元素个数
就用np.multiply(*a.size())