tensor和numpy的转换

共享内存的,转换很快:

import torch
import numpy as np

# tensor转换为numpy
b = a.numpy()

# numpy转换为tensor
d = torch.from_numpy(c)

tensor转换为list:

# 先转换成numpy再转换成list
list = tensor.numpy().tolist()

发布于 2022-06-06 15:48