TypeError: randint(): argument 'size' (position 2) must be tuple of ints, not int

TypeError: randint(): argument 'size' (position 2) must be tuple of ints, not int¶

randint 메서드를 사용할때 rand메서드와 같이 텐서의 사이즈를 입력하면 아래와 같은 에러가 발생합니다.¶

In [1]:

import torch
torch.randint(1,2)

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
/tmp/ipykernel_2374966/3186067014.py in <module>
      1 import torch
----> 2 torch.randint(1,2)

TypeError: randint(): argument 'size' (position 2) must be tuple of ints, not int

randint 메서드에 정수의 범위와 사이즈를 아래처럼 입력하면 에러 없이 사용가능 합니다.¶

In [2]:

# 0 부터 10 사이 정수를 (1,2) 사이즈로 생성해 주세요
torch.randint(0,10, (1,2))

Out[2]:

tensor([[8, 5]])

[참고] rand 매서드는 (1,2) 사이즈를 입력해도 에러가 발생하지 않습니다.¶

In [3]:

torch.rand(1,2)

Out[3]:

tensor([[0.6253, 0.5698]])