tf.searchsorted | TensorFlow v2.0.0 (original) (raw)

tf.searchsorted

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

Searches input tensor for values on the innermost dimension.

View aliases

Compat aliases for migration

tf.compat.v1.searchsorted

tf.searchsorted(
    sorted_sequence, values, side='left', out_type=tf.dtypes.int32, name=None
)

A 2-D example:

  sorted_sequence = [[0, 3, 9, 9, 10],
                     [1, 2, 3, 4, 5]]
  values = [[2, 4, 9],
            [0, 2, 6]]

  result = searchsorted(sorted_sequence, values, side="left")

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

  result = searchsorted(sorted_sequence, values, side="right")

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

Args
sorted_sequence	N-D Tensor containing a sorted sequence.
values	N-D Tensor containing the search values.
side	'left' or 'right'; 'left' corresponds to lower_bound and 'right' to upper_bound.
out_type	The output type (int32 or int64). Default is tf.int32.
name	Optional name for the operation.

Returns
An N-D Tensor the size of values containing the result of applying either lower_bound or upper_bound (depending on side) to each value. The result is not a global index to the entire Tensor, but the index in the last dimension.

Raises
ValueError	If the last dimension of sorted_sequence >= 2^31-1 elements. If the total size of values exceeds 2^31 - 1 elements. If the first N-1 dimensions of the two tensors don't match.