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()