torch.rand()

torch.rand함수는 [0,1) 구간에서 균등 분포를 따르는 난수로 채워진 텐서를 반환

 

parameter

size -> 말 그대로 어떤 모양으로 반환할지, 리스트나 튜플로도 가능

out: 출력 텐서를 지정

여기서 Out이 좀 새로웠는데, 새로운 텐서를 생성하는 것이 아닌 기존에 있는 곳에 보낼 수 있다는 것임

예를 들면

import torch

# 기존 텐서 생성
c = torch.empty(3, 4)

# 새로운 텐서를 생성하지 않고 c에 결과 저장
a = torch.rand(3, 4)
torch.add(a, 3, out=c)

 

requires_grad(연산 저장 및 추적)와 pin_memory(DRAM을 거치지 않고 VRAM으로 데이터를 보내는)도 있음.