In this case we have to use the tensor.permute() attribute with PyTorch. tensor_np_P = np.transpose(tensor_np, (2,1,3,4,0)) tensor_pt_P ...
確定! 回上一頁