方法:
torch.numel(x)
作用: 返回给定数组的元素个数
官方示例:
>>> a = torch.randn(1, 2, 3, 4, 5) >>> torch.numel(a) 120 >>> a = torch.zeros(4,4) >>> torch.numel(a) 16
方法:
torch.numel(x)
作用: 返回给定数组的元素个数
官方示例:
>>> a = torch.randn(1, 2, 3, 4, 5) >>> torch.numel(a) 120 >>> a = torch.zeros(4,4) >>> torch.numel(a) 16