pytorch获取张量的shape

可以使用张量的shape属性来获取张量的形状，例如：

import torch

x = torch.randn(3, 4, 5)
print(x.shape)

输出结果为：

torch.Size([3, 4, 5])

这表示x是一个3维张量，第一维有3个元素，第二维有4个元素，第三维有5个元素。