Spec-Zone.ru › TensorFlow 2.3

tf.searchsorted

Просмотреть исходный код на GitHub

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

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

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

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

tf.compat.v1.searchsorted

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

Пример с 2-мерным тензором:

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 размерности значений, содержащий результат применения либо нижней границы, либо верхней (в зависимости от заданного параметра side) к каждому значению. Результат не является глобальным индексом всего Tensor, а индексом в последнем измерении.
Исключения
ValueError Если последнее измерение sorted_sequence >= 2^31-1 элементов. Если общий размер значений превышает 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.3/api_docs/python/tf/searchsorted

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API