tf.raw_ops.ApproxTopK

Приблизительно возвращает значения min/max k и их индексы для входного операнда.

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

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

Подробнее см. в руководстве по миграции.

tf.compat.v1.raw_ops.ApproxTopK

tf.raw_ops.ApproxTopK(
    input,
    k,
    reduction_dimension=-1,
    recall_target=0.95,
    is_max_k=True,
    reduction_input_size_override=-1,
    aggregate_to_topk=True,
    name=None
)

Подробные сведения об алгоритме см. на странице https://arxiv.org/abs/2206.14286. В настоящее время эта операция оптимизирована только для TPU.

Аргументы
`input`	A `Tensor`. Должен иметь один из следующих типов: `half`, `bfloat16`, `float32`. Массив для поиска. Должен иметь размерность не менее 1 и тип с плавающей точкой
`k`	An `int` that is `>= 0`. Задает количество значений min/max-k.
`reduction_dimension`	An optional `int`. По умолчанию — `-1`. Целочисленная размерность, вдоль которой выполняется поиск. По умолчанию: -1.
`recall_target`	An optional `float`. По умолчанию — `0.95`. Целевое значение полноты для приближения. Диапазон: (0,1]
`is_max_k`	An optional `bool`. По умолчанию — `True`. Если значение равно true, вычисляет max-k; в противном случае — min-k.
`reduction_input_size_override`	An optional `int`. По умолчанию — `-1`. Если задано положительное значение, оно переопределяет размер, определяемый с помощью `input[reduction_dim]`, для вычисления полноты. Этот параметр полезен, когда заданный `input` представляет собой только подмножество общего вычисления в SPMD или распределенных конвейерах, где истинный размер входных данных нельзя определить по форме `input`.
`aggregate_to_topk`	An optional `bool`. По умолчанию — `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/raw_ops/ApproxTopK