使用numpy的ndarray处理数据非常有用
在基于PyTorch的模型中也经常使用,不可避免地要将jsdmt与多维数组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中共享内存