pytorch求 tensor的元素个数 | python中*args和**kwargs的含义

在学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())