Tensor를 .npy 파일로 저장, 로드

Pytorch Tensor 자료형을 .npy로 저장하고, 불러오고, Tensor로 다시 변형!!
그냥 자주 써서 정리해보았다 !!

Tensor를 numpy로 변경

a_np = a.numpy( )
이 때, tensor의 device type이 'cuda'이면 'cpu'로 바꿔줘야 한다.
b_np = b.cpu( ).numpy( )

npy파일로 local에 저장

numpy 배열 'b'를 경로에 저장
np.save('/data/npy_file', b)

npy파일 불러오기

np_load = np.load('/data/npy_file.npy')

numpy Array⇒Tensor

#device type이 'cuda'인 tensor 완성
result = torch.from_numpy(np_load).to('cuda')

csr matrix to pytorch sparse tenser

참고 코드 링크를 아래에 첨부 !!
https://gist.github.com/aesuli/319d71707a5ee96086aa2439b87d4e38