더북(TheBook)

torch.chunk는 텐서를 쪼갤 때 사용하는 함수이며 파라미터는 다음과 같습니다.

ⓐ 첫 번째 파라미터: 텐서를 몇 개로 쪼갤지 설정합니다.

ⓑ 두 번째 파라미터: 어떤 차원을 기준으로 쪼갤지를 결정합니다. dim=1이므로 열 단위로 텐서를 분할하겠다는 의미입니다.

uniform()은 난수를 위해 사용합니다. 난수 사용은 uniform() 외에도 randint(), random() 등이 있는데, 사용 방법은 다음과 같습니다.

from random import *

ri = randint(1, 10) ------ 1부터 10 사이의 임의의 정수
print(ri)

rd = random() ------ 0부터 1 사이의 임의의 실수(float)
print(rd)

ui = uniform(1, 10) ------ 1부터 10 사이의 임의의 실수(float)
print(ui)

rr = randrange(1, 10, 2) ------ 1부터 10 사이를 2씩 건너뛴 임의의 정수( 1, 3, 5, 7, 9)
print(rr)
신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.