tensorflow::ops::MaxPoolWithArgmax

#include <nn_ops.h>

Выполняет max pooling для входных данных и возвращает как максимальные значения, так и индексы.

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

Индексы в 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), даже если используется дополнение и математически правильный ответ находится за пределами диапазона (отрицательное значение или слишком большое значение). Это ошибка, но её трудно исправить безопасным способом с сохранением обратной совместимости, особенно из-за преобразования в сплющенный вид.

Аргументы:

• scope: объект Scope
• input: тензор размерности 4 с формой [batch, height, width, channels]. Входные данные, к которым применяется pooling.
• ksize: размер окна для каждого измерения входного тензора.
• strides: шаг скользящего окна для каждого измерения входного тензора.
• padding: тип используемого алгоритма дополнения.

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

• include_batch_in_index: указывает, нужно ли включать измерение пакета в сплющенный индекс argmax.

Возвращает:

• Output output: тензор с результатом max pooling.
• Output argmax: размерность 4. Сплющенные индексы максимальных значений, выбранных для каждого выходного элемента.

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

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