tf.nn.max_pool_with_argmax

Просмотреть исходный код на GitHub

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

tf.nn.max_pool_with_argmax(
    input, ksize, strides, padding, data_format='NHWC',
    output_dtype=tf.dtypes.int64, include_batch_in_index=False, name=None
)

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

Аргументы
`input`	A `Tensor`. Должен быть одного из следующих типов: `float32`, `float64`, `int32`, `uint8`, `int16`, `int8`, `int64`, `bfloat16`, `uint16`, `half`, `uint32`, `uint64`. 4-мерный с формой `[batch, height, width, channels]`. Входные данные для пулингового преобразования.
`ksize`	Целое число или список `ints` длиной `1`, `2` или `4`. Размер окна для каждого измерения входного тензора.
`strides`	Целое число или список `ints` длиной `1`, `2` или `4`. Шаг скользящего окна для каждого измерения входного тензора.
`padding`	A `string` из: `"SAME", "VALID"`. Тип алгоритма заполнения.
`data_format`	Необязательный `string`, должен быть установлен в `"NHWC"`. По умолчанию `"NHWC"`. Указывает формат данных входных и выходных данных.
`output_dtype`	Необязательный `tf.DType` из: `tf.int32, tf.int64`. По умолчанию `tf.int64`. Тип данных возвращаемого тензора argmax.
`include_batch_in_index`	Необязательный `boolean`. По умолчанию `False`. Включить ли размер пакетной обработки в сглаженном индексе `argmax`.
`name`	Имя операции (необязательно).

Возвращаемые значения
Кортеж объектов `Tensor` (выход, argmax).
`output`	A `Tensor`. Имеет тот же тип, что и `input`.
`argmax`	A `Tensor` типа `output_dtype`.

© 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.4/api_docs/python/tf/nn/max_pool_with_argmax