tf.raw_ops.ApproxTopK
Возвращает минимальные/максимальные k-значения и их индексы входного операнда приближенным способом.
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, который является >= 0. Указывает количество минимальных/максимальных k-значений. |
reduction_dimension | Необязательный int. По умолчанию -1. Целевая размерность для поиска. Значение по умолчанию: -1. |
recall_target | Необязательный float. По умолчанию 0.95. Цель отклика для аппроксимации. Диапазон (0,1] |
is_max_k | Необязательный bool. По умолчанию True. Если True, вычисляет максимальные k-значения; в противном случае вычисляет минимальные k-значения. |
reduction_input_size_override | Необязательный int. По умолчанию -1. Если задано положительное значение, оно переопределяет размер, определяемый input[reduction_dim] для оценки отклика. Этот параметр полезен, когда заданный input является только подмножеством всего вычисления в SPMD или распределенных конвейерах, где истинный размер входных данных не может быть отложен формой input. |
aggregate_to_topk | Необязательный 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
