如何在tensorflow中使用gather nd

Question

我有一个大小为（B，N）的数组“状态”，其中有true或false指示每个批次中每个点的状态，还有另一个大小为（B，N，2）的“位置”数组，其中position（ b，n）指示点在网格中的位置。

status = tf.constant(np.random.randint(2, size=(B, N)))
# Each point has a status where status[b, n] indicates 
# status of the point
status = tf.cast(status, tf.bool)
# position(b, n) = (x, y) where (x, y) is coordinate of the point in the grid. 
# Each point falls into a cell. 
position = tf.constant(np.random.randint(10, size=(B, N, 2)))

如何在tensorflow中计算2d网格的每个单元中的假和真数？我希望输出的大小为（b，X，Y），其中X和Y是网格的大小。