Pytorch之扩充tensor的操作

在 PyTorch 中，我们可以使用以下方法来扩充 tensor 的操作。

方法1：使用 torch.unsqueeze()

我们可以使用 torch.unsqueeze() 函数来扩充 tensor 的维度。

import torch

# 定义一个 2x3 的 tensor
x = torch.tensor([[1, 2, 3], [4, 5, 6]])

# 使用 torch.unsqueeze() 扩充维度
x = torch.unsqueeze(x, dim=0)

# 输出 tensor 的形状
print(x.shape)

在这个示例中，我们首先定义了一个 2x3 的 tensor x。然后，我们使用 torch.unsqueeze() 函数将 tensor x 扩充为一个 1x2x3 的 tensor。在使用 torch.unsqueeze() 函数时，我们需要指定要扩充的维度 dim。

方法2：使用 torch.expand()

我们可以使用 torch.expand() 函数来扩充 tensor 的形状。

import torch

# 定义一个 2x3 的 tensor
x = torch.tensor([[1, 2, 3], [4, 5, 6]])

# 使用 torch.expand() 扩充形状
x = torch.expand(x, (3, 2, 3))

# 输出 tensor 的形状
print(x.shape)

在这个示例中，我们首先定义了一个 2x3 的 tensor x。然后，我们使用 torch.expand() 函数将 tensor x 扩充为一个 3x2x3 的 tensor。在使用 torch.expand() 函数时，我们需要指定要扩充的形状。注意，要扩充的形状必须符合一定的规则，例如，要扩充的维度的大小必须为 1，或者要扩充的维度的大小必须与原来的大小相同。