Python 是否在Tensorflow上的GPU上执行非最大值抑制？

Tensorflow为非最大值抑制提供了这种方法，我想知道Tensorflow是否提供了GPU支持？

答案是否。如果您尝试：

tf.image.non_max_suppression 

你会看到：

boxes = np.random.uniform(size=(100, 4))
scores = np.random.uniform(size=(100,))

bx = tf.constant(boxes, dtype=tf.float32)
sc = tf.constant(scores, dtype=tf.float32)

with tf.device("gpu:0"):
    indx = tf.image.non_max_suppression(bx, sc, 1000)

不

    NotFoundError: No registered 'NonMaxSuppressionV3' OpKernel for GPU devices compatible with node {{node NonMaxSuppressionV3}} = NonMaxSuppressionV3[T=DT_FLOAT](dummy_input, dummy_input, dummy_input, dummy_input, dummy_input)
    .  Registered:  device='CPU'; T in [DT_HALF]
  device='CPU'; T in [DT_FLOAT]
 [Op:NonMaxSuppressionV3]