Blog Full Notice
back to main page

최대 1 분 소요

motivation: pytorch의 torch.randint에 대해서 알아보자.

1. randint(end: int , size=(3,1): tuple) -> tensor[[1],[8],[9]])

import torch
tensor = torch.randint(10, (3,1))
tensor

하면 tensor[[1],[8],[9]]) 가 나온다. 0부터(start 생략하면 0부터) end까지의 정수 중에 uniform하게, 나오는 것.

2. randint(end: int , size=(3,): tuple) -> tensor[1,8,9])

3. randint(end: int , size=(3,0): tuple) -> tensor[])

아무것도 안나온다.

댓글남기기