[PyTorch] torch.nn.init

torch.nn.init에는 tensor에 초깃값을 주기 위해 필요한 함수들이 저장되어 있다. torch.nn.init.uniform_(tensor) x 변수에 [10,1] 모양의 tensor을 생성하고 이 텐서의 값들을 init.uniform_() 함수를 통해서 -10부터 10까지 uniform(균등) 하게 초기화한다. 그럼 -10 ~ 10까지의 숫자들이 무작위로 들어가게 된다. PyTorch 공식 문서를 보면 uniform distribution U(a, b)에서 가져온 값으로 input Tensor를 채운다. 매개변수 - tensor : n 차원의 tensor를 넣어주면 된다. - a는 균일 분포의 하한 - b는 균일 분포의 상한 torch.nn.init.normal_(tensor) init.uniform_과 유사하다. N(mean, std^2)을 따르는 정규 분포에서의 값을 input tens..........

원문링크 : [PyTorch] torch.nn.init