tensor to numpy / numpy to tensor

jaeha_lee·2021년 5월 12일
0

딥러닝을 사용하다보면 맨날 까먹어서 찾아보고는 하는데, 이 기회에 정리해보고자 한다.

  1. Numpy -> Tensor
output = torch.Tensor(input) #(새 메모리 할당)input은 Numpy Output은 Tensor
output = torch.from_numpy(input) #(원래 메모리 할당)input은 Numpy Output은 Tensor
  1. Tensor -> Numpy
output = np.array(input) #input은 Tensor /  Output은 Numpy

0개의 댓글