코드:



결과:



설명(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

+ Recent posts