unsqueeze函数

unsqueeze函数是PyTorch中的一个函数，用于在指定维度上增加一个维度。例如，假设一个tensor的shape为(3,4)，我们可以使用unsqueeze函数在第一维度上增加一个维度，使其变成(1,3,4)。

具体用法如下：

import torch

t = torch.tensor([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])

# 在第一维度上增加一个维度
t.unsqueeze(0)

这将返回一个新的tensor，其shape为(1,3,4)。要在其他维度上增加维度，只需要修改参数即可。