PyTorch 1~2강) Introduction / Basic

한량·2021년 8월 17일
0

[U-stage] PyTorch for AI

목록 보기
1/6
post-thumbnail

Functions

View / Reshape의 차이

  • View: 두 텐서가 동일한 데이터를 공유
    그래서 연속성 제약 조건(contiguous)이 있음
  • Reshape: 데이터 공유x, 제약조건x

Squeeze

  • dim = 1인 axis 제거

dot / mm / matmul 차이

  • matmul은 각 batch마다 mm을 해준것과 같다

torch.gather

PyTorch 필수과제를 참고

profile
놀고 먹으면서 개발하기

0개의 댓글