unsqueeze函数的使用方法，我需要增加一个维度

`unsqueeze` 函数可以用来增加一个维度。其函数原型如下：

unsqueeze(input, dim)

其中，`input` 表示需要增加维度的张量，`dim` 表示需要在哪个维度上增加维度。

举例来说，如果有一个形状为 `(3, 4)` 的张量 `x`，想要在第二个维度上增加一个维度，使得其形状变为 `(3, 1, 4)`，则可以使用如下代码：

import torch

x = torch.randn(3, 4)
x = x.unsqueeze(1)
print(x.shape)

输出结果为：

torch.Size([3, 1, 4])

其中，`x.unsqueeze(1)` 表示在第二个维度上增加一个维度。