Spec-Zone.ru › TensorFlow 1.15

tf.searchsorted

View source on GitHub

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

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

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

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

tf.compat.v1.searchsorted, `tf.compat.v2.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 размерности значений, содержащий результат применения либо нижней границы, либо верхней границы (в зависимости от стороны) к каждому значению. Результат не является глобальным индексом к всему 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/r1.15/api_docs/python/tf/searchsorted

Spec-Zone.ru

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