tf.approx_top_k

Возвращает приближенные минимальные/максимальные k значений и их индексов для входного операнда.

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

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

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

tf.compat.v1.approx_top_k

tf.approx_top_k(
    input: Annotated[Any, TV_ApproxTopK_T],
    k: int,
    reduction_dimension: int = -1,
    recall_target: float = 0.95,
    is_max_k: bool = True,
    reduction_input_size_override: int = -1,
    aggregate_to_topk: bool = True,
    name=None
)

См. https://arxiv.org/abs/2206.14286 для получения подробной информации об алгоритме. Этот оператор оптимизирован только на TPU.

Аргументы
`input`	A `Tensor`. Должен быть одного из следующих типов: `half`, `bfloat16`, `float32`. Массив для поиска. Должен быть по крайней мере одномерным с плавающей точкой
`k`	Целое число, указывающее количество минимальных/максимальных k значений.
`reduction_dimension`	Необязательное целое число. По умолчанию - `-1`. Индекс измерения, по которому нужно искать. Значение по умолчанию: -1.
`recall_target`	Необязательное число с плавающей точкой. По умолчанию - `0.95`. Цель полноты для приближения. Значение от (0,1]
`is_max_k`	Необязательное логическое значение. По умолчанию - `True`. Если True, вычисляются максимальные k; в противном случае вычисляются минимальные k.
`reduction_input_size_override`	Необязательное целое число. По умолчанию - `-1`. При задании положительного значения оно переопределяет размер, определяемый `input[reduction_dim]` для оценки полноты. Этот параметр полезен, когда заданный `input` представляет собой только подмножество всего вычисления в SPMD или распределённых конвейерах, где истинный размер входных данных не может быть определён по форме `input`.
`aggregate_to_topk`	Необязательное логическое значение. По умолчанию - `True`. При значении True агрегируются приближённые результаты до top-k. При значении False возвращаются приближённые результаты. Количество приближённых результатов определяется реализацией и не меньше заданного `k`.
`name`	Имя операции (необязательно).

Возвращаемые значения
Кортеж объектов `Tensor` (значения, индексы).
`values`	A `Tensor`. Имеет тот же тип, что и `input`.
`indices`	A `Tensor` типа `int32`.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/approx_top_k