torch.unsqueeze是什么意思

torch.unsqueeze是一个PyTorch库函数，它的作用是对输入的Tensor进行维度扩展，将指定维度的大小增加1。具体来说，它能够在指定的维度上添加一个新的维度，使得原有的Tensor的维度增加1。该函数的输入是一个Tensor，以及一个整数值dim，表示要在哪个维度上增加新的维度。该函数的返回值是一个新的Tensor，它的维度比原有的Tensor多了一个。