Searches input tensor for values on the innermost dimension.
Aliases:
tf.compat.v1.searchsorted
tf.compat.v2.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-DTensor
containing a sorted sequence.values
: N-DTensor
containing the searchvalues
.side
: 'left' or 'right'; 'left' corresponds to lower_bound and 'right' to upper_bound.out_type
: The output type (int32
orint64
). Default istf.int32
.name
: Optionalname
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 firstN-1
dimensions of the two tensors don't match.