[pytorch] tensor변형 방법들 - view, reshape, transpose, permute

문지우·2024년 9월 4일

torch.view() & torch.reshape()

공통점

: 둘 다 tensor 모양 바꾸는데에 사용됨

x = torch.rand(2, 3, 4)  # [2, 3, 4] 
y = x.view(2, -1)  # [2, 12] 

x = torch.rand(2, 3, 4)  # [2, 3, 4] 
x = x.reshape(2, -1) # [2, 12]

# -1 : 알아서 계산하라는 의미(자동 계산)

차이점

is_contigious : tensor 값들 순차적인지
view - contigious 하지 않는 tensor에 적용 불가
reshape - contigious 하지 않는 tensor에 적용 가능


https://sanghyu.tistory.com/3

profile
대왕 감자의 성장 일기,,,👩‍🌾

0개의 댓글