언어/파이썬

4차원 tensor axis에 따른 np.sum() 정리

M.랄라 2022. 12. 30. 17:31

일단, 아래에서 말하는 행, 렬은 모두 위의 그림을 기준으로 말하는 것임.

 

1) axis = 0 인 경우

큰 행렬 입장에서의 열을 더하는 경우

 

2) axis =1 인 경우

큰 행렬 입장에서 행을 더하는 경우

 

3) axis =2 인 경우

작은 행렬 입장에서 열을 더하는 경우

 

4) axis =3 인 경우

작은 행렬 입장에서 행을 더하는 경우

 

 

참고 유투브 : https://youtu.be/gWyVWXFDrW0 (수원대 한경훈교수님) 밑바닥부터 시작하는 딥러닝

'언어 > 파이썬' 카테고리의 다른 글

python 3차원 tensor, 4차원 tensor  (0) 2022.12.30