_get_ref_points(H, W, B, dtype, device) displacement = (q_grid.reshape(B * self.n_groups, H * W, 2).unsqueeze(2) - pos.reshape(B ...
確定! 回上一頁