파이썬 Pytorch 텐서에서 numpy 배열로

torch.Size ([4, 3, 966, 1296]) 크기의 pytorch Tensor가 있습니다.

다음 코드를 사용하여 numpy 배열로 변환하고 싶습니다.

imgs = imgs.numpy () [:, ::-1, :, :]

누구든지이 코드가 무엇을하는지 설명해 주시겠습니까?

 

해결 방법

 

변환하려는 텐서에는 4 개의 차원이 있습니다.

[:, ::-1, :, :] 

: 는 첫 번째 차원이 그대로 복사되고 변환되어야 함을 의미하며 세 번째 및 네 번째 차원도 동일합니다.

::-1 은 두 번째 축의 경우 축이 반전됨을 의미합니다.

 

참조 페이지 https://stackoverflow.com/questions/49768306