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

输出:
pytorch求 tensor的元素个数 | python中*args和**kwargs的含义_第1张图片

所以* 表示是解包,将size的[3,4]解包为3,4。之后再进行相乘操作,打包解包详解参见https://www.cnblogs.com/mo-nian/p/11842422.html
好了,以后求tensor元素个数就用np.multiply(*a.size())

你可能感兴趣的:(pytorch,numpy,pytorch,python,深度学习)