tensor某个维度上做扩张 自身重复方法

在 PyTorch 中，可以使用 `torch.unsqueeze()` 方法对张量的指定维度进行扩张，例如：

import torch

x = torch.tensor([[1, 2], [3, 4]])
y = torch.unsqueeze(x, dim=0)  # 在第0维进行扩张
print(y)

输出为：

tensor([[[1, 2],
         [3, 4]]])

如果要在某个维度上重复张量自身，可以使用 `torch.repeat()` 方法，例如：

import torch

x = torch.tensor([[1, 2], [3, 4]])
y = x.repeat(2, 3)  # 在第0维重复2次，在第1维重复3次
print(y)

输出为：

tensor([[1, 2, 1, 2, 1, 2],
        [3, 4, 3, 4, 3, 4],
        [1, 2, 1, 2, 1, 2],
        [3, 4, 3, 4, 3, 4]])

其中参数 `2` 和 `3` 分别表示在第0维和第1维上重复的次数。