тензорный поток:: опс:: ТопК

#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)

Публичные атрибуты

indices
operation
values

Публичные статические функции

Sorted (bool x)

Структуры

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
)