Pytorch中ndarray tensor list互轉

1.ndarray->tensor :

b=torch.from_numpy(a)

2.tensor->ndarray:

b=a.numpy()

'''
但這么寫會報錯……
RuntimeError: Can't call numpy() on Variable that requires grad. Use var.detach().numpy() instead.
'''
# 修改為
b=a.detach().numpy()

'''
gpu上的tensor不能直接轉為numpy
'''
b=a.cpu().numpy()

3.ndarray->list

b=a.tolist()

4.list->ndarray

b=numpy.array(a)