rquires_grad1 torch.rand() torch.rand함수는 [0,1) 구간에서 균등 분포를 따르는 난수로 채워진 텐서를 반환 parametersize -> 말 그대로 어떤 모양으로 반환할지, 리스트나 튜플로도 가능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으로 데이터를 보내는)도 있음. 2024. 8. 11. 이전 1 다음