pytorch中的unsqueeze()函数解析

0.doc文档

doc文档是这么说的:
pytorch中的unsqueeze()函数解析
这里简单的译一下。

1.func:

返回一个新的tensor,这个tensor 在指定的位置被插入了一个大小为1的新维度
这个返回的tensor 和之前的 tensor 有着相同的数据

2.parameter

2.1

tensor: 就是需要被处理的那个tensor

2.2

dim 值的范围在 [-input.dim()-1, input.dim()+1 ),注意是左闭右开区间。 如果dim是负数的话,那么传入 unsqueeze()函数中 dim真正实际值是 dim+input.dim() + 1