Torch.Tensor can generate 'nan' elements

nn.Parameter(torch.Tensor(batch, h_dim))

위의 tensor는 종종 nan element를 발생시킨다. 따라서 다음과 같이 tensor 생성에 사용하는것을 추천한다.

 

nn.Parameter(torch.rand(batch, h_dim))