pytorch查看张量维度

在PyTorch中，你可以使用`torch.Tensor`对象的`.shape`属性来查看张量的维度。这个属性返回一个包含张量每个维度大小的元组。例如，如果你有一个二维张量（矩阵），它的形状可能是`(2, 3)`，表示有两行三列。

如果你想获取更具体的维度信息，你也可以使用`torch.ndim()`或`len(tensor)`方法，它们都会返回张量的维度数。例如：

import torch

# 创建一个张量
tensor = torch.randn(3, 4, 5)
print("Tensor shape:", tensor.shape)  # 输出: (3, 4, 5)
print("Number of dimensions:", tensor.ndim)  # 输出: 3