tensorflow ValueError：Shape必须为1级，但是为2级

Question

import tensorflow as tf
x = [[1,2,3],[4,5,6]]
y = [0,1]
z = [1,2]
x = tf.constant(x)
y = tf.constant(y)
z = tf.constant(z)
m = x[y,z]

我期望的是m = [2,6]

我可以通过theano或numpy获得结果。我如何使用tensorflow获得结果？

Answer 1

您可能希望使用tf.gather_nd

   slices = tf.gather_nd(x, [y, z])

希望这会有所帮助。