如何将CuPy数组转换为tens

Question

我正在使用CuPy在GPU上生成一些ground truth+网络输入数据，我想通过标准API将这些数据发送到TensorFlow网络： 进给量={网络。标签：标签，network.network_输入：networkInput}

但是，TensorFlow 1.15期望的是NumPy而不是CuPy，我得到一个错误： 文件“/lib/python3.6/site-packages/numpy/core/数值.py“，第538行，在阿萨雷 返回数组（a，dtype，copy=False，order=order） 值错误：object数组方法未生成数组

正在尝试通过labels将CuPy数组转换为NumPy数组数字阵列（标签，数据类型=np.浮动) 我得到一个错误：ValueError:object数组方法没有生成数组

有没有一种方法可以将CuPy数据传递给tensorflow？在

Answer 1

要将CuPy数组（在GPU上）复制到NumPy数组（在CPU上），请使用cupy.asnumpy(arr)或arr.get()。在

https://docs-cupy.chainer.org/en/latest/reference/ndarray.html#conversion-to-from-numpy-arrays