모든 연산에서 dim = # 의 방향은 다음과 같다. 열, 행 .. 순이다.

 

1. view

 

reshape를 하는 함수이다.

{tensor}.view(dimension = []) 꼴로 shape를 바꿀 수 있다.

 

2. squeeze, unsqueeze

 

특정 위치(dim)에 1인 차원을 추가하거나 삭제한다. 데이터 모양 때문에 보통 사용한다.

 

3. In-place Operation

 

기존 함수들끝에 _를 붙이면 덮어쓰기 연산이 되는 것들이 몇 있다.

'머신 러닝 및 파이썬 > 딥러닝 기초 , 파이토치' 카테고리의 다른 글

parameter 확인하는 법  (0) 2021.01.03
Multi-layer 예시 [Diabetes Logistics]  (0) 2020.12.29
Linear Regression  (0) 2020.12.29
Supervised Learning  (0) 2020.02.18

+ Recent posts