torch.shape 和 torch.size() 输出结果相同。

x = torch.tensor([[0, 2],[3,4],[9,8]])
print(x.shape)

在这里插入图片描述

x = torch.tensor([[[0, 2],[0,8],[2,7]],[[2,5],[9,3],[7,3]]])
print(x)
print(x.shape)

在这里插入图片描述

x = torch.tensor([0,1,2,4,5,6,2])
print(x.shape)

在这里插入图片描述

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐