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