tensorflow::ops::ApproxTopK

#include <nn_ops.h>

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

Краткое описание

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

Аргументы:

• scope: объект Scope
• input: массив для поиска. Должен иметь не менее 1 измерения и быть вещественного типа
• k: указывает количество элементов min/max-k.

Необязательные атрибуты (см. Attrs):

• reduction_dimension: целочисленное измерение, по которому выполняется поиск. По умолчанию: -1.
• recall_target: целевое значение полноты для приближения. Диапазон в (0,1]
• is_max_k: если значение true, вычисляет max-k; в противном случае вычисляет min-k.
• reduction_input_size_override: если задано положительное значение, оно заменяет размер, определяемый input[reduction_dim] для вычисления полноты. Этот параметр полезен, если заданный input является лишь подмножеством общего вычисления в конвейерах SPMD или распределённых конвейерах, где истинный размер входных данных нельзя определить по форме input.
• aggregate_to_topk: если значение true, объединяет приближённые результаты в top-k. Если значение false, возвращает приближённые результаты. Количество приближённых результатов определяется реализацией и не меньше указанного k.

Возвращает:

• Output значения: значения min/max k по измерению reduction_dimension операнда input. Измерения совпадают с измерениями операнда input, за исключением reduction_dimension: если aggregate_to_topk имеет значение true, размерность редукции равна k; в противном случае она не меньше k, размер которой определяется реализацией.
• Output индексы: индексы values по измерению reduction_dimension операнда input.

Открытые атрибуты

::tensorflow::Output indices

Operation operation

::tensorflow::Output values

Открытые функции

 ApproxTopK(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  int64 k
)

 ApproxTopK(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  int64 k,
  const ApproxTopK::Attrs & attrs
)

Открытые статические функции

Attrs AggregateToTopk(
  bool x
)

Attrs IsMaxK(
  bool x
)

Attrs RecallTarget(
  float x
)

Attrs ReductionDimension(
  int64 x
)

Attrs ReductionInputSizeOverride(
  int64 x
)