ABOUT ME

-

Today
-
Yesterday
-
Total
-
  • transpose와 permute
    카테고리 없음 2023. 3. 22. 16:27

    permute()와 transpose()는 유사한 방식으로 작동한다.

    transpose()는 딱 두 개의 차원을 맞교환할 수 있다.

    그러나 permute()는 모든 차원들을 맞교환할 수 있다.

    x = torch.rand(16, 32, 3)
    y = x.tranpose(0, 2)  # [3, 32, 16]
    z = x.permute(2, 1, 0)  # [3, 32, 16]

     

Designed by Tistory.