【冰糖Python】PyTorch tensor 和 Numpy ndarray 互相转换

使用 numpy 里的 ndarray 处理数据非常方便

在基于 PyTorch 的模型中也会经常用到，难免会将张量 tensor 和 多维数组 ndarray 进行互相转换

牢记以下安全转换方式很有必要

1、ndarray 转 tensor

torch.from_numpy(ndarray)

将张量 numpy.ndarray 转换为 tensor 类型，返回的 tensor 和 ndarray 在 cpu 中共享内存，修改其一同时改变

当然，使用 torch.tensor() 直接构造 tensor 也可以

2、tensor 转 ndarray

tensor.numpy()

假设有一个 tensor，则调用 tensor.numpy() 可将其转换为 numpy.ndarray，在 cpu 中共享内存