목록Slice and Index (1)

HAZEL

[ PyTorch 01. ] Basic, Operations, View, Slice and Index, Compile, AutoGrad

01. PyTorch Basic : 타입만 다를 뿐 numpy 와 유사하게 동작한다. import numpy as np import torch 1. 기본 arange print(np.arange(9)) # [0 1 2 3 4 5 6 7 8] print(torch.arange(9)) # tensor([0, 1, 2, 3, 4, 5, 6, 7, 8]) 2. shape # shape 를 알수 있음 nums = torch.arange(6) nums.shape 3. type type(nums) 4. numpy 로 변환 # 넘파이 변환이 가능함 nums.numpy() 5. reshape # numpy 와 유사하게 reshape 를 할 수 있음 nums.reshape(2,3) 6. rand # 랜덤으로 3,3 텐서 ..
DATA ANALYSIS/ML & DL 2021. 4. 29. 12:50
이전 Prev 1 Next 다음