pytorch 的 nelement()
pytorch中的 nelement() 可以统计 tensor (张量) 的元素的个数。输入代码：环境 pytorch 1.0 ，Python 3.6import torchx = torch.randn(size = (4,3,5,6))x = torch.tensor([[1.], [2], [3]])print(x.nelement())结果是：3如果有误，...
·
- pytorch中的 nelement() 可以统计 tensor (张量) 的元素的个数。
- 输入代码：环境 pytorch 1.0 ，Python 3.6
import torch
x = torch.randn(size = (4,3,5,6))
x = torch.tensor([[1.], [2], [3]])
print(x.nelement())- 结果是：
3
如果有误，欢迎指正！
有“AI”的1024 = 2048，欢迎大家加入2048 AI社区
更多推荐
6
0- 0
已为社区贡献2条内容
所有评论(0)