mm = torch.randn(2,3,4)
for i in range(2):
n = mm[i]
print(n.shape)
torch.Size([3, 4])
torch.Size([3, 4])
注意此时 n的shape不是[1,3, 4]，而是[3, 4]
mm = torch.randn(2,3,4)
for i in range(2):
n = mm[i]
print(n.shape)
torch.Size([3, 4])
torch.Size([3, 4])
注意此时 n的shape不是[1,3, 4]，而是[3, 4]
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
