тензорный поток:: опс:: Макспулвисаргмакс
#include <nn_ops.h>
Выполняет максимальное объединение входных данных и выводит как максимальные значения, так и индексы.
Краткое содержание
Индексы в argmax
сглаживаются, так что максимальное значение в позиции [b, y, x, c]
становится сглаженным индексом: (y * width + x) * channels + c
если include_batch_in_index
имеет значение False; ((b * height + y) * width + x) * channels + c
если include_batch_in_index
имеет значение True.
Возвращаемые индексы всегда имеют размер [0, height) x [0, width)
до выравнивания, даже если используется заполнение и математически правильный ответ находится снаружи (отрицательный или слишком большой). Это ошибка, но ее трудно исправить безопасным способом с обратной совместимостью, особенно из-за сглаживания.
Аргументы:
- область: объект области.
- ввод: 4-D с формой
[batch, height, width, channels]
. Вход в пул окончен. - ksize: Размер окна для каждого измерения входного тензора.
- шаги: шаг скользящего окна для каждого измерения входного тензора.
- дополнение: тип используемого алгоритма заполнения.
Необязательные атрибуты (см. Attrs
):
- include_batch_in_index: включать ли измерение пакета в плоский индекс
argmax
.
Возврат:
-
Output
результат: максимальный объединенный выходной тензор. -
Output
аргументмакс: 4-D. Сглаженные индексы максимальных значений, выбранных для каждого выхода.
Конструкторы и деструкторы | |
---|---|
MaxPoolWithArgmax (const :: tensorflow::Scope & scope, :: tensorflow::Input input, const gtl::ArraySlice< int > & ksize, const gtl::ArraySlice< int > & strides, StringPiece padding) | |
MaxPoolWithArgmax (const :: tensorflow::Scope & scope, :: tensorflow::Input input, const gtl::ArraySlice< int > & ksize, const gtl::ArraySlice< int > & strides, StringPiece padding, const MaxPoolWithArgmax::Attrs & attrs) |
Публичные атрибуты | |
---|---|
argmax | |
operation | |
output |
Публичные статические функции | |
---|---|
IncludeBatchInIndex (bool x) | |
Targmax (DataType x) |
Структуры | |
---|---|
tensorflow:: ops:: MaxPoolWithArgmax:: Attrs | Необязательные установщики атрибутов для MaxPoolWithArgmax . |
Публичные атрибуты
аргументмакс
::tensorflow::Output argmax
операция
Operation operation
выход
::tensorflow::Output output
Общественные функции
Макспулвисаргмакс
MaxPoolWithArgmax( const ::tensorflow::Scope & scope, ::tensorflow::Input input, const gtl::ArraySlice< int > & ksize, const gtl::ArraySlice< int > & strides, StringPiece padding )
Макспулвисаргмакс
MaxPoolWithArgmax( const ::tensorflow::Scope & scope, ::tensorflow::Input input, const gtl::ArraySlice< int > & ksize, const gtl::ArraySlice< int > & strides, StringPiece padding, const MaxPoolWithArgmax::Attrs & attrs )
Публичные статические функции
Инклудбатчининдекс
Attrs IncludeBatchInIndex( bool x )
Таргмакс
Attrs Targmax( DataType x )