Возвращает значения min/max k и их индексы входного операнда приближенным образом.
Подробности алгоритма см. на https://arxiv.org/abs/2206.14286. В настоящее время эта операция оптимизирована только для TPU.
Вложенные классы
сорт | ApproxTopK.Options | Дополнительные атрибуты для ApproxTopK |
Публичные методы
статический ApproxTopK.Options | агрегатToTopk (логическое выражениеагрегатToTopk) |
static <T расширяет число> ApproxTopK <T> | create (Область действия , ввод операнда <T>, длинный k, параметры... параметры) Фабричный метод для создания класса, обертывающего новую операцию ApproxTopK. |
Вывод <Целое число> | индексы () Индексы «значений» в «reduction_dimension» операнда «input». |
статический ApproxTopK.Options | isMaxK (логическое значение isMaxK) |
статический ApproxTopK.Options | RecallTarget (Float RecallTarget) |
статический ApproxTopK.Options | ReductionDimension (Длинное сокращениеDimension) |
статический ApproxTopK.Options | сокращениеInputSizeOverride (длинное сокращениеInputSizeOverride) |
Выход <Т> | ценности () Значения min/max k вдоль `reduction_dimension` операнда `input`. |
Унаследованные методы
Публичные методы
общедоступный статический ApproxTopK.OptionsагрегатToTopk ( логический агрегатToTopk)
Параметры
агрегаттотопк | Если это правда, результаты агрегируются с точностью до top-k. Если значение false, возвращает приблизительные результаты. Количество приблизительных результатов определяется реализацией и больше, равно указанному `k`. |
---|
public static ApproxTopK <T> create ( Область действия, ввод операнда <T>, Long k, параметры... параметры)
Фабричный метод для создания класса, обертывающего новую операцию ApproxTopK.
Параметры
объем | текущий объем |
---|---|
вход | Массив для поиска. Должен быть как минимум 1-D плавающего типа. |
к | Указывает количество min/max-k. |
параметры | содержит значения необязательных атрибутов |
Возврат
- новый экземпляр ApproxTopK
общедоступный статический ApproxTopK.Options isMaxK (логическое значение isMaxK)
Параметры
isMaxK | Если это правда, вычисляет max-k; в противном случае вычисляет min-k. |
---|
общедоступный статический ApproxTopK.Options RecallTarget (Float RecallTarget)
Параметры
отзывTarget | Вспомните цель аппроксимации. Диапазон в (0,1] |
---|
общедоступный статический ApproxTopK.Options сокращениеDimension (длинное сокращениеDimension)
Параметры
уменьшениеразмерность | Целочисленное измерение, по которому осуществляется поиск. По умолчанию: -1. |
---|
общедоступный статический ApproxTopK.Options сокращениеInputSizeOverride (длинное сокращениеInputSizeOverride)
Параметры
сокращениеInputSizeOverride | Если установлено положительное значение, оно переопределяет размер, определенный `input[reduction_dim]` для оценки отзыва. Эта опция полезна, когда данный «вход» является лишь подмножеством общих вычислений в SPMD или распределенных конвейерах, где истинный размер ввода не может быть отложен формой «входа». |
---|
общедоступные выходные значения <T> ()
Значения min/max k вдоль `reduction_dimension` операнда `input`. Размерность такая же, как у операнда `input`, за исключением `reduction_dimension`: когда `aggregate_to_topk` имеет значение true, размерность уменьшения равна `k`; в противном случае оно больше равно `k`, где размер определяется реализацией.