tensor.shape()怎么用

`tensor.shape()` 用于获取张量（tensor）的形状信息，返回一个元组（tuple）类型的值，元组中的每个元素表示张量在相应维度上的长度。

以下是使用示例：

import torch

# 创建一个形状为 (2, 3) 的张量
x = torch.Tensor([[1, 2, 3], [4, 5, 6]])

# 获取张量形状信息
shape = x.shape

print(shape)  # 输出：torch.Size([2, 3])
print(shape[0])  # 输出：2
print(shape[1])  # 输出：3

其中，`torch.Size` 是 `torch` 库中的一个类，用于表示张量的形状，它本质上是一个元组类型。你可以通过下标访问元组中的元素，以获取张量在相应维度上的长度。