코드:
결과:
설명(https://pytorch.org/docs/stable/_modules/torch/nn/functional.html)
텐서의 각 행이나 열을 정규화할 수 있다. 위 코드에서 a의 경우 3*4 텐서인데
dim=0을 파라미터로 할 경우 첫번째 dimension 즉 column을 정규화하고
dim=1으로 할 경우 두번째 dimension 즉 row를 정규화 한다