pytorch输出张量的阶数和形状

PyTorch输出张量的阶数和形状可以使用张量的dim和size方法来获取。

- 张量的阶数：可以使用dim()方法获取张量的阶数，返回一个整数，表示张量的维度数。
- 张量的形状：可以使用size()或shape属性获取张量的形状，返回一个元组，表示张量在每个维度上的大小。

例如，对于一个名为tensor的张量，可以使用以下代码获取它的阶数和形状：

import torch

tensor = torch.tensor([[1, 2], [3, 4], [5, 6]])

# 获取张量的阶数
print(tensor.dim())  # 输出 2

# 获取张量的形状
print(tensor.size())  # 输出 torch.Size([3, 2])
print(tensor.shape)  # 输出 torch.Size([3, 2])

在这个例子中，张量tensor是一个2维张量，形状为3行2列。