使用 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 中共享内存

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐