tensorflow::ops::MaxPoolWithArgmax
#include <nn_ops.h>
Выполняет максимальное пулинговое преобразование над входом и выводит как максимальные значения, так и индексы.
Summary
Индексы в 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]. Вход для пулинга. - ksize: Размер окна для каждой размерности входного тензора.
- strides: Шаг скользящего окна для каждой размерности входного тензора.
- padding: Тип алгоритма заполнения.
Необязательные атрибуты (см. Attrs):
- include_batch_in_index: Включать ли размерность батча в сглаженный индекс
argmax.
Возвращает:
-
Outputoutput: Тензор с максимальным пулингом. -
Outputargmax: 4-мерный. Сглаженные индексы максимальных значений, выбранных для каждого выхода.
| Конструкторы и деструкторы | |
|---|---|
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. |
Публичные атрибуты
argmax
::tensorflow::Output argmax
operation
Operation operation
output
::tensorflow::Output output
Публичные функции
MaxPoolWithArgmax
MaxPoolWithArgmax( const ::tensorflow::Scope & scope, ::tensorflow::Input input, const gtl::ArraySlice< int > & ksize, const gtl::ArraySlice< int > & strides, StringPiece padding )
MaxPoolWithArgmax
MaxPoolWithArgmax( const ::tensorflow::Scope & scope, ::tensorflow::Input input, const gtl::ArraySlice< int > & ksize, const gtl::ArraySlice< int > & strides, StringPiece padding, const MaxPoolWithArgmax::Attrs & attrs )
Публичные статические функции
IncludeBatchInIndex
Attrs IncludeBatchInIndex( bool x )
Targmax
Attrs Targmax( DataType x )
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/cc/class/tensorflow/ops/max-pool-with-argmax