(bs, self.na, self.no, ny, nx).permute(0, 1, 3, 4, 2).contiguous() if not self.training:...(bs, self.na, self.no, ny, nx).permute(0, 1, 3, 4, 2).contiguous() if not self.training:...(bs, self.na, self.no, ny, nx).permute(0, 1, 3, 4, 2).contiguous() x[i + self.nl] = self.m2...(0, 1, 3, 4, 2).contiguous() if not self.training: # inference if self.grid...(bs, self.na, self.no, ny, nx).permute(0, 1, 3, 4, 2).contiguous() if not self.training: