У меня есть массив 'status' размера (B, N) с true или false, указывающим состояние каждой точки в каждом пакете, и другой массив 'position' размера (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)))
Как подсчитать количество ложных и истинных значений в каждой ячейке 2-й сетки в тензорном потоке?Я ожидаю выходных данных размера (b, X, Y), где X и Y - размеры сетки.