파이토치 정수 랜덤 생성

파이토치 정수 랜덤 생성

# low 에서 high 사이 정수를 size 크기 만큼 생성
torch.randint(low, high, size)


# 예시 : torch.randint(0, 10, [2,3])
# out : tensor([[9, 9, 0],
#               [5, 2, 0]])

 

 

[참고]

# size 는 참고 용입니다.
size = [3,2,4]

torch.rand(size) # 0~1 사이 난수 생성(균일분포)
torch.randn(size) # 0~1 사이 난수 생성(가우시안분포)