pytorch中如何扩充张量维数

您可以使用 `unsqueeze` 方法来扩展张量的维度。`unsqueeze` 方法接受一个参数，即要扩展的维度的索引。以下是一个示例代码：

import torch

# 创建一个 2x3 的张量
x = torch.tensor([[1, 2, 3], [4, 5, 6]])

# 在索引为 0 的维度上扩展张量的维度
expanded_x = x.unsqueeze(0)

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

在上面的示例中，通过调用 `unsqueeze(0)`，我们在索引为 0 的维度上扩展了张量 `x` 的维度。这将在原始张量的外部添加一个新的维度，大小为 1。