pytorch permute维度转换方法
permute
prediction=input.view(bs,self.num_anchors, self.bbox_attrs,in_h,in_w).permute(0,1,3,4,2).contiguous()
转置:
importtorch x=torch.linspace(1,9,steps=9).view(3,3) b=x.permute(1,0) print(b) print(b.permute(1,0))
以上这篇pytorchpermute维度转换方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持毛票票。