TopK

публичный финальный класс TopK

Находит значения и индексы k крупнейших элементов для последнего измерения.

Если входные данные являются вектором (ранг-1), находит k крупнейших записей в векторе и выводит их значения и индексы в виде векторов. Таким образом, «values[j]» — это «j»-я по величине запись в «input», а ее индекс — «indices[j]».

Для матриц (соответственно входные данные более высокого ранга) вычисляет верхние элементы `k` в каждой строке (соответственно вектор по последнему измерению). Таким образом,

значения.shape = index.shape = input.shape[:-1] + [k]

Если два элемента равны, первым отображается элемент с меньшим индексом.

Вложенные классы

сорт ТопК.Опции Дополнительные атрибуты для TopK

Константы

Нить OP_NAME Название этой операции, известное основному движку TensorFlow.

Публичные методы

статический <T расширяет TNumber > TopK <T>
create ( Область действия , ввод операнда <T>, операнд <TInt32> k, параметры... параметры)
Фабричный метод для создания класса, обертывающего новую операцию TopK.
Вывод < TInt32 >
индексы ()
Индексы «значений» в последнем измерении «входа».
статический TopK.Options
отсортировано (логическая сортировка)
Выход <Т>
ценности ()
Наибольшие элементы k на каждом последнем размерном срезе.

Унаследованные методы

Константы

общедоступная статическая финальная строка OP_NAME

Название этой операции, известное основному движку TensorFlow.

Постоянное значение: «ТопКВ2».

Публичные методы

public static TopK <T> create ( Область действия, ввод операнда <T>, операнд < TInt32 > k, параметры... параметры)

Фабричный метод для создания класса, обертывающего новую операцию TopK.

Параметры
объем текущий объем
вход 1-D или выше с последним измерением не ниже «k».
к 0-Д. Количество верхних элементов, которые нужно искать по последнему измерению (вдоль каждой строки для матриц).
параметры содержит значения необязательных атрибутов
Возврат
  • новый экземпляр ТопК

публичный вывод < TInt32 > индексы ()

Индексы «значений» в последнем измерении «входа».

общедоступная статическая сортировка TopK.Options (логическая сортировка)

Параметры
отсортированный Если установлено значение true, результирующие элементы `k` будут отсортированы по значениям в порядке убывания.

общедоступные выходные значения <T> ()

Наибольшие элементы k на каждом последнем размерном срезе.