tensorflow::ops::MaxPoolWithArgmax
#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) до сглаживания, даже если используется заполнение, и математически правильный ответ находится вне диапазона (либо отрицательный, либо слишком большой). Это ошибка, но исправить её сложно, сохраняя обратную совместимость, особенно из-за сглаживания.
Аргументы:
- scope: Объект Scope
- input: 4-мерный с формой
[batch, height, width, channels]. Вход для пулингового усреднения. - ksize: Размер окна для каждого измерения входного тензора.
- strides: Шаг скользящего окна для каждого измерения входного тензора.
- padding: Тип алгоритма заполнения.
Необязательные атрибуты (см. Attrs):
- include_batch_in_index: Включать ли размерность пакетного (batch) в сглаженный индекс
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
выход
::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 )
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/cc/class/tensorflow/ops/max-pool-with-argmax