tf.searchsorted

Ищет значения во входном тензоре по внутреннему измерению.

Просмотр псевдонимов

Псевдонимы для миграции

См. Руководство по миграции для получения дополнительной информации.

tf.compat.v1.searchsorted

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

Пример с двумерным тензором:

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]]

Аргументы
`sorted_sequence`	N-мерный `Tensor` содержащий отсортированную последовательность.
`values`	N-мерный `Tensor` содержащий значения для поиска.
`side`	'left' или 'right'; 'left' соответствует нижней границе, а 'right' - верхней.
`out_type`	Тип результата (`int32` или `int64`). По умолчанию `tf.int32`.
`name`	Необязательное имя операции.

Возвращаемое значение

Возвращаемое значение
N-мерный `Tensor` того же размера, что и values, содержащий результат применения lower_bound или upper_bound (в зависимости от параметра side) к каждому значению. Результат не является глобальным индексом всего `Tensor`, а индексом в последнем измерении.

N-мерный Tensor того же размера, что и values, содержащий результат применения lower_bound или upper_bound (в зависимости от параметра side) к каждому значению. Результат не является глобальным индексом всего Tensor, а индексом в последнем измерении.

Исключения
`ValueError`	Если последнее измерение `sorted_sequence >= 2^31-1` элементов. Если общий размер values превышает `2^31 - 1` элементов. Если первые `N-1` измерения двух тензоров не совпадают.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/searchsorted