pytorch的unsqueeze和unsqueeze_的使用

pytorch的unsqueeze和unsqueeze_的使用,及in_place操作

unsqueeze

unsqueeze就是用来扩展维度的, 先来看下unsqueeze的参数, torch.unsqueeze(input, dim, out=None), 扩维的范围是[-input.dim() - 1, input.dim() + 1), 如果是传入的dim是负数,那么dim实际数值是dim = dim + input.dim() + 1.
其实上代码就能看清楚了, c最开始的维度数只有一维, 然后在0维度扩展后,维度数量变成2, 所谓的0,1维度其实和pandas的0,1维是一致的,0代表行扩展,1代表列扩展.
我们也可以看到所谓输入的负数其实和输入正数是等价的, 可以自己试下输入的范围不在扩维的范围内,将会报错

unsqueeze_

unsqueeze_和unsqueeze实现一样的功能, 区别在于unsqueeze_是in_place操作,即unsqueeze不会对使用unsqueeze的tensor进行改变,想要获取unsqueeze后的值必须赋予个新值,unsqueeze_则会对自己改变