Находит значения и индексы 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.
Публичные методы
public static TopK <T> create ( Область действия, ввод операнда <T>, операнд < TInt32 > k, параметры... параметры)
Фабричный метод для создания класса, обертывающего новую операцию TopK.
Параметры
объем | текущий объем |
---|---|
вход | 1-D или выше с последним измерением не ниже «k». |
к | 0-Д. Количество верхних элементов, которые нужно искать по последнему измерению (вдоль каждой строки для матриц). |
параметры | содержит значения необязательных атрибутов |
Возврат
- новый экземпляр ТопК
общедоступная статическая сортировка TopK.Options (логическая сортировка)
Параметры
отсортированный | Если установлено значение true, результирующие элементы `k` будут отсортированы по значениям в порядке убывания. |
---|