list/numpy/pytorch tensor相互转换
List 与 Numpy相互转换
-
numpy to list
import numpy as np if __name__ == '__main__': n = np.arange(5) # fisrt method #a = n.tolist() # second method a = list(n) print(a)
-
list to numpy
import numpy as np if __name__ == '__main__': a = [1,2,3,4] n = np.array(a) print(n)
List 与 Tensor 相互转换
-
numpy to tensor
import numpy as np import torch if __name__ == '__main__': n = np.arange(5) t = torch.from_numpy(n) print(t)
-
tensor to numpy
import numpy as np import torch if __name__ == '__main__': t = torch.arange(5) n = t.numpy() print(n)