tf.compat.v2.math.bincount

Подсчитывает количество вхождений каждого значения в целочисленный массив.

tf.compat.v2.math.bincount(
    arr, weights=None, minlength=None, maxlength=None, dtype=tf.dtypes.int32,
    name=None
)

Если minlength и maxlength не заданы, возвращает вектор длины tf.reduce_max(arr) + 1, если arr не пусто, и длины 0 в противном случае. Если weights не None, то индекс i выходного значения хранит сумму значений в weights в каждом индексе, где соответствующее значение в arr равно i.

values = tf.constant([1,1,2,3,2,4,4,5])
tf.math.bincount(values) #[0 2 2 1 2 1]

Длина вектора = Максимальное значение в векторе values равно 5. Добавление 1, что составляет 6, будет длиной вектора.

Каждое значение ячейки в выходном значении указывает количество вхождений конкретного индекса. Здесь индекс 1 в выходном значении имеет значение 2. Это указывает на то, что значение 1 встречается дважды в values.

values = tf.constant([1,1,2,3,2,4,4,5])
weights = tf.constant([1,5,0,1,0,5,4,5])
tf.math.bincount(values, weights=weights) #[0 6 0 1 9 5]

Ячейка будет инкрементироваться соответствующим весом вместо 1. Здесь индекс 1 в выходном значении имеет значение 6. Это сумма весов, соответствующих значению в values.