tf.raw_ops.UpperBound

Применяет upper_bound(sorted_search_values, values) к каждой строке.

tf.raw_ops.UpperBound(
    sorted_inputs,
    values,
    out_type=tf.dtypes.int32,
    name=None
)

Каждый набор строк с одинаковым индексом в (sorted_inputs, values) обрабатывается независимо. Результирующая строка эквивалентна вызову np.searchsorted(sorted_inputs, values, side='right').

Результат — это не глобальный индекс во всём Tensor, а только индекс в последнем измерении.

Пример для двумерного случая: sorted_sequence = [[0, 3, 9, 9, 10], [1, 2, 3, 4, 5]] values = [[2, 4, 9], [0, 2, 6]]

result = UpperBound(sorted_sequence, values)

result == [[1, 2, 4], [0, 2, 5]]