반응형
cuda로 올리기
-
머신러닝: 파이썬 Pytorch 사용하기!머신러닝 2020. 11. 9. 19:05
1. torch.rand(a, b) : 0 ~ 1사이의 값을 갖는 a x b 텐서생성 import torch x = torch.rand(5, 3) x tensor([[0.2107, 0.1415, 0.2014], [0.3747, 0.4420, 0.9662], [0.8733, 0.0158, 0.6805], [0.8282, 0.7829, 0.2063], [0.9674, 0.3998, 0.0489]]) tensor형태로 출력 2. tensor 값 지정해주기 x = torch.tensor([[0,1,1,2], [2,2,3,1]]) x tensor([[0, 1, 1, 2], [2, 2, 3, 1]]) 3. torch.zeros(a, b) : 0 값을 가진 a x b의 tensor 생성 torch.ones(a, b)..