TensorFlow的random_poisson仅在CPU上运行

Question

我正在尝试让TensorFlow的Random_Poisson函数在我的GPU上运行；假设这个TensorFlow源页面有一个函数TestCPUGPUMatch，该函数在CPU和GPU上运行时比较Random_Poisson的输出，那么这似乎是可能的。但是，在使用代码进行测试时：

import tensorflow as tf

with tf.Session() as sess:
    with tf.device("/gpu:0"):
        test = sess.run(tf.random_poisson(1.0, [], dtype=tf.float64))
print(test)

我得到了错误：

谢了！

Answer 1

没有针对随机泊松的GPU内核。在random_poisson_op.cc中，只注册CPU内核。

如果查看链接的TestCPUGPUMatch代码，它会调用self.test_session(use_gpu=true,...），如果可能，它会尝试在GPU上运行ops。在暗地里，test_session使用allow_soft_placement来执行此操作，如果不可能在GPU上运行操作，则返回到CPU。所以测试实际上在CPU上运行op。

[旁白：为什么我们有一个真正不做任何事情的测试？TestCPUGPUMatch似乎是许多不同随机操作的测试（请参见Test_Random_Ops.py)，而RandomPoisson的实现者可能正是出于这个原因才添加了它。似乎将它命名为TestCPUGPUMatchifGPUimplementationExists...]

可以通过github问题提交一个GPU内核的特性请求，或者提交一个带有实现的PR。