tensorflow::ops::TopK

#include <nn_ops.h>

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

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

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

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

values.shape = indices.shape = input.shape[:-1] + [k]

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

Аргументы:

• scope: объект Scope
• input: одномерный массив или массив более высокого ранга с последним измерением размером не менее k.
• k: скаляр. Количество наибольших элементов, которые нужно найти вдоль последнего измерения (в каждой строке матриц).

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

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

Возвращает:

• Output values: k наибольших элементов вдоль каждого среза последнего измерения.
• Output indices: индексы values в последнем измерении input.

Открытые атрибуты

::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 IndexType(
  DataType x
)

Attrs Sorted(
  bool x
)