pytorch

Tensor 생성 리스트(list) 또는 NumPy 배열(ndarray)로부터 Tensor 생성 ( list도 가능하다) import torch import numpy as np # 리스트로부터 Tensor 생성 list_data = [1, 2, 3] tensor_data = torch.tensor(list_data) # NumPy 배열로부터 Tensor 생성 ndarray_data = np.array([[1, 2], [3, 4]]) tensor_data = torch.tensor(ndarray_data) Tensor Operation 기본적으로 numpy 연산과 같다. data = [[3,5,20],[10,5,50],[1,5,10]] x_data = torch.tensor(data) x_data[1:,..