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

Находит значения и индексы 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 .

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

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