tf.keras.layers.MaxPool2D 最大池化层 示例

tf.keras.layers.MaxPool2D 最大池化层 示例

import tensorflow as tf
import numpy as np

inputs = np.random.randint(  # 随机生成整数
    low=1,  # 最小整数（包括）
    high=10,  # 最大整数（不包括）
    size=(4, 28, 28, 3)  # 形状
)

inputs = tf.convert_to_tensor(inputs) # numpy 数组转 tensor
print(inputs.shape)

outputs = tf.keras.layers.MaxPooling2D(
    pool_size=(2, 2),  # 通过在输入的每个通道上获取输入窗口（大小由 定义）上的最大值，沿其空间维度（高度和宽度）对输入进行降采样
    strides=(1, 1),  # 步长
    padding='valid'
)(inputs)
# output_shape = math.floor((input_shape - pool_size) / strides) + 1
print(outputs.shape)

(4, 27, 27, 3)