pytorch中的数据类型之间的转换数据类型之间的转换
trunc函数类型1. numpy转list
list = list()
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()

版权声明：本站内容均来自互联网，仅供演示用，请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系QQ:729038198，我们将在24小时内删除。