тензорный поток:: опс:: ТопК
#include <nn_ops.h>
Находит значения и индексы k
крупнейших элементов для последнего измерения.
Краткое содержание
Если входные данные являются вектором (ранг-1), находит k
крупнейших элементов вектора и выводит их значения и индексы в виде векторов. Таким образом, values[j]
— это j
-я по величине запись в input
, а ее индекс — indices[j]
.
Для матриц (соответственно входные данные более высокого ранга) вычисляет первые k
записей в каждой строке (соответственно вектор по последнему измерению). Таким образом,
values.shape = indices.shape = input.shape[:-1] + [k]
Если два элемента равны, первым отображается элемент с меньшим индексом.
Аргументы:
- область: объект области.
- ввод: 1-D или выше с последним измерением не менее
k
. - к: 0-Д. Количество верхних элементов для поиска по последнему измерению (вдоль каждой строки для матриц).
Необязательные атрибуты (см. Attrs
):
- sorted: если true, то полученные
k
элементов будут отсортированы по значениям в порядке убывания.
Возврат:
-
Output
значения:k
крупнейших элементов в каждом последнем размерном срезе. -
Output
индексы: индексыvalues
в последнем измеренииinput
.
Конструкторы и деструкторы | |
---|---|
TopK (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input k) | |
TopK (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input k, const TopK::Attrs & attrs) |
Структуры | |
---|---|
tensorflow::ops::TopK::Attrs | Дополнительные установщики атрибутов для TopK . |
Публичные атрибуты
индексы
::tensorflow::Output indices
операция
Operation operation
ценности
::tensorflow::Output values
Общественные функции
ТопК
TopK( const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input k )
ТопК
TopK( const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input k, const TopK::Attrs & attrs )
Публичные статические функции
Отсортировано
Attrs Sorted( bool x )