Pytorch 学习2

1、torch.range(num1，num2)

莫名其妙，pytorch这个range函数和python内嵌的不一样，是[ start , end ]

第一，必须有两个参数

第二，包括第二个参数

 

2、tensor.unsqueeze(dim )

在第dim维外增加一个维度，比如原来是（2，3），假如dim是0，输出是[1,2,3]，假如dim是1，输出是[2,1,3]，假如是2，输出是[2,3,1]

3、tensor.squeeze(dim)

可以减掉维度，但是只有维度为1时才会去掉，其他不会变。