코드:



결과:



설명(https://pytorch.org/docs/stable/_modules/torch/nn/functional.html)


텐서의 각 행이나 열을 정규화할 수 있다. 위 코드에서 a의 경우 3*4 텐서인데 


dim=0을 파라미터로 할 경우 첫번째 dimension 즉 column을 정규화하고


dim=1으로 할 경우 두번째 dimension 즉 row를 정규화 한다

'딥러닝 > 백과사전' 카테고리의 다른 글

백과사전 카테고리  (0) 2018.06.20

연구하면서 언제든 다시 필요할만한 각종 스킬들,정보들 정리해놓는 곳

'딥러닝 > 백과사전' 카테고리의 다른 글

텐서 정규화(normalize)하기  (0) 2018.06.20

+ Recent posts