可以快速上手的开发文档
permuteprediction=input.view(bs,self.num_anchors,self.bbox_attrs,in_h,in_w).permute(0,1,3,4,2).contiguous()转置:impo..
permute(dims)将tensor的维度换位。参数:参数是一系列的整数,代表原来张量的维度。比如三..