tf.raw_ops.MaxPoolWithArgmax

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

Просмотреть псевдонимы

Псевдонимы совместимости для миграции

Подробнее см. в руководстве по миграции.

tf.compat.v1.raw_ops.MaxPoolWithArgmax

tf.raw_ops.MaxPoolWithArgmax(
    input,
    ksize,
    strides,
    padding,
    Targmax=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`	`Tensor`. Должен иметь один из следующих типов: `float32`, `float64`, `int32`, `uint8`, `int16`, `int8`, `int64`, `bfloat16`, `uint16`, `half`, `uint32`, `uint64`. Тензор ранга 4 с формой `[batch, height, width, channels]`. Входные данные для пулизации.
`ksize`	Список `ints` длиной `>= 4`. Размер окна для каждого измерения входного тензора.
`strides`	Список `ints` длиной `>= 4`. Шаг скользящего окна для каждого измерения входного тензора.
`padding`	`string` из: `"SAME", "VALID"`. Тип используемого алгоритма дополнения.
`Targmax`	Необязательный `tf.DType` из: `tf.int32, tf.int64`. По умолчанию используется `tf.int64`.
`include_batch_in_index`	Необязательный `bool`. По умолчанию — `False`. Определяет, включать ли измерение пакета в уплощённый индекс `argmax`.
`name`	Имя операции (необязательно).

Возвращает
Кортеж объектов `Tensor` (output, argmax).
`output`	`Tensor`. Имеет тот же тип, что и `input`.
`argmax`	`Tensor` типа `Targmax`.

© 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/api_docs/python/tf/raw_ops/MaxPoolWithArgmax