Функция nn.max_pool1d

Функция nn.max_pool1d применяет операцию максимального пулинга к одномерным данным, например, к временным рядам или последовательностям. Первым параметром функция принимает входной тензор формы [batch, length, channels]. Вторым параметром передается размер окна пулинга ksize, третьим - шаг скольжения окна strides. Четвертым параметром задается тип дополнения padding (значение 'VALID' или 'SAME'), пятым - формат данных data_format. Функция возвращает тензор, содержащий максимальные значения в каждом окне.

Синтаксис

tf.nn.max_pool1d(input, ksize, strides, padding, data_format='NWC', name=None)

Пример

Давайте применим одномерный максимальный пулинг к тензору с одним каналом. Входной тензор имеет форму [1, 5, 1], окно пулинга равно 2, шаг равен 1:

import tensorflow as tf

t = tf.constant([[[1], [2], [3], [4], [5]]], dtype=tf.float32)
res = tf.nn.max_pool1d(t, ksize=2, strides=1, padding='VALID')
print(res)

Результат выполнения кода:

tf.Tensor(
[[[2.]
  [3.]
  [4.]
  [5.]]], shape=(1, 4, 1), dtype=float32)

Пример

Давайте применим пулинг с дополнением 'SAME' и шагом 2 к тому же тензору:

import tensorflow as tf

t = tf.constant([[[1], [2], [3], [4], [5]]], dtype=tf.float32)
res = tf.nn.max_pool1d(t, ksize=2, strides=2, padding='SAME')
print(res)

Результат выполнения кода:

tf.Tensor(
[[[2.]
  [4.]
  [5.]]], shape=(1, 3, 1), dtype=float32)

Пример

Давайте применим одномерный максимальный пулинг к тензору с двумя каналами. Входной тензор имеет форму [1, 5, 2], окно пулинга равно 3, шаг равен 2:

import tensorflow as tf

t = tf.constant([[[1, 2], [3, 4], [5, 6], [7, 8], [9, 10]]], dtype=tf.float32)
res = tf.nn.max_pool1d(t, ksize=3, strides=2, padding='VALID')
print(res)

Результат выполнения кода:

tf.Tensor(
[[[5. 6.]
  [9. 10.]]], shape=(1, 2, 2), dtype=float32)

Смотрите также

функцию max_pool,
которая выполняет максимальный пулинг для двумерных данных
функцию max_pool2d,
которая выполняет двумерный максимальный пулинг
функцию avg_pool,
которая выполняет усредняющий пулинг
функцию conv1d,
которая выполняет одномерную свертку