pytorch如何在tensor的前方增加维度

在PyTorch中，你可以使用`unsqueeze()`函数来在Tensor的前方增加维度。`unsqueeze()`函数的参数是要插入的维度的索引，从0开始计数。例如，在一个形状为(2,3)的Tensor中，在第0个维度上插入一个大小为1的新维度，可以这样实现：

import torch

x = torch.randn(2, 3)
print(f"x shape: {x.shape}")  # 输出：x shape: torch.Size([2, 3])

x = x.unsqueeze(0)
print(f"x shape after unsqueeze: {x.shape}")  # 输出：x shape after unsqueeze: torch.Size([1, 2, 3])

在这个例子中，我们使用了`unsqueeze()`函数将一个形状为(2,3)的Tensor在第0个维度上插入了一个大小为1的新维度，得到了一个形状为(1,2,3)的Tensor。