tensorflow::ops::ApproxTopK

#include <nn_ops.h>

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

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

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

Аргументы:

• scope: Объект Scope
• input: Массив для поиска. Должен быть по крайней мере 1-мерным и иметь тип с плавающей точкой
• k: Указывает количество минимальных/максимальных k значений.

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

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

Возвращаемые значения:

• Output значения: Минимальные/максимальные 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
)