pytorch中的數據類型之間的轉換

數據類型之間的轉換

1. numpy轉list

list = ndarray.tolist()

2. tensor轉numpy

ndarray = tensor.numpy()
# 注意gpu上的tensor不能直接轉為numpy，應先放回cpu中
ndarray = tensor.cpu().numpy()   # 注意requires_grad為True的值應該調用.detach()方法

3. tensor轉list

# 先轉為numpy再轉為list
list = tensor.numpy().tolist()

4. CPU tensor和GPU tensor之間的轉換

# CPU--->GPU
data = data.cuda()
# GPU--->CPU
data = data.cpu()