torch.numel()用法

torch.numel()用来统计tensor中元素的个数

#输入
a=torch.tensor([1,2,3,4])
print(torch.numel(a))

#输出
4


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