Tensor 복습

김보현·2024년 8월 8일
0

PyTorch

목록 보기
6/7

목표

헷갈렸던 method를 다시 한번 정리해본다!

Tensor 확장

  • repeat 메서드, expand 메서드
    - repeat 메서드는 실제로 데이터를 복제하므로 메모리 사용량이 늘어난다.
    - expand는 메모리 효율적으로 크기를 확장한다.

변수 h에 크기가 (2,3)인 2-D Tensor가 저장되어 있을 때, h.expand(4,3)은 크기가 (4,3)인 2-D Tensor로 가상으로 확장한다.

변수 h에 2-D Tensor가 저장되어 있을 때, h.repeat(1,2)는 dim=0축으로 1번, dim=1축으로 2번 반복하여 Tensor의 크기를 확장하는 코드다.

  • expand 메서드는 메모리 효율성이 좋다.
    expand 메서드는 메모리를 복제하지 않고 동일한 데이터를 가리키면서 크기를 확장하기 때문에 메모리 효율성이 좋다.

  • repeat 메서드는 Tensor의 차원의 크기가 1이 아닌 경우에도 사용할 수 있다.

profile
Fall in love with Computer Vision

0개의 댓글