pytorch 遇到的小问题总结

一、

transpose：是permute的精简版本。transpose(dim1, dim2)只能将两个维度进行互换

y = y.permute(2,3,1,4) -- 按照2,3,1,4维进行重排列

二、

对两个variable进行concat操作，按道理实现方式是c = torch.cat([a, b], dim=0)，但提示错误

TypeError: cat received an invalid combination of arguments - got (tuple, int), but expected one of:

• (sequence[torch.cuda.FloatTensor] tensors)
• (sequence[torch.cuda.FloatTensor] tensors, int dim)
  didn’t match because some of the arguments have invalid types: (tuple, int)

解决办法：根据提示刚开始以为是cat不接受tuple作为输入，然而真正的问题在于a和b的type不一样，比如可能出现a是torch.cuda.DoubleTensor而b是torch.cuda.FloatTensor，因此，将a和b转换为相同的type即可。

如 a : torch.FloatTensor  和  b : torch.cuda.FloatTensor     此时需要增加  a = a.cuda()  

    

三、

pytorch 得到Variable  的值    Variable.data