tf.raw_ops.UpperBound  |  TensorFlow v2.16.1 (original) (raw)

tf.raw_ops.UpperBound

Stay organized with collections Save and categorize content based on your preferences.

Applies upper_bound(sorted_search_values, values) along each row.

View aliases

Compat aliases for migration

SeeMigration guide for more details.

tf.compat.v1.raw_ops.UpperBound

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

Each set of rows with the same index in (sorted_inputs, values) is treated independently. The resulting row is the equivalent of callingnp.searchsorted(sorted_inputs, values, side='right').

The result is not a global index to the entireTensor, but rather just the index in the last dimension.

A 2-D example: 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]]