【代码可运行】
【调试过程】
【视频】非常详见的讲解
print('深码云端')
print('--------------------')
##本文件主题 :
# # tensor size ,shape,numel
import torch
a = torch.randn( 3, 4, 5)
print(a)
print(a.size())
print(a.shape)
count = torch.numel(a)
print(count)
if __name__ == '__main__':
print('--------------------')
print('深码云端')
运行结果
深码云端
--------------------
tensor([[[ 1.6214, 1.8099, -1.2566, 0.1755, 0.1683],
[-0.7764, -0.7178, 1.3427, -0.7126, 3.0117],
[ 1.4555, -0.7081, 0.5703, -0.5007, -1.0897],
[-0.3636, -1.0235, 0.3895, -0.8908, 1.1101]],
[[-1.7623, 0.1895, 1.2988, -0.2174, 1.0175],
[ 0.1021, -0.9948, -1.8234, -0.2803, -0.6421],
[-0.3464, -0.1567, -0.4136, 1.8940, -0.7885],
[ 0.0132, -0.7587, -0.4736, -1.1177, 0.3447]],
[[-0.9167, 0.1505, -0.5630, 0.4265, 2.0090],
[ 1.0367, 0.0429, -0.5484, -0.1096, -0.6836],
[-1.3655, 0.5248, 0.5915, 1.2664, -1.7323],
[ 0.0636, 1.3731, -0.8730, 2.0249, -0.4639]]])
torch.Size([3, 4, 5])
torch.Size([3, 4, 5])
60
--------------------
深码云端
进程已结束,退出代码0
pytorch, size ,shape,numel