TensorFlow.js tf.layers.maxPooling3d() 函数

简介

tf.layers.maxPooling3d() 是 TensorFlow.js 中用于创建 3D 最大池化层的函数。最大池化层是卷积神经网络 (CNN) 中常用的一种操作，它用于减小输入数据的空间维度，同时保留最显著的特征。该函数基于 WebGL 技术实现，可以在浏览器中进行高效的前向传播计算。

语法

tf.layers.maxPooling3d(config)

参数

• config (对象): 配置项对象，包含以下属性：

  • poolSize (数字数组或数字): 指定池化窗口的大小。如果是数字，则三个维度上的窗口大小相同；如果是数字数组，应指定每个维度上的窗口大小。例如，[2, 2, 2] 表示在三个维度上分别使用 2x2x2 大小的池化窗口。

  • strides (数字数组或数字): 指定池化窗口在每个维度上的步幅大小。默认值是 undefined，表示步幅大小与池化窗口大小相同。如果是数字，则三个维度上的步幅大小相同；如果是数字数组，应指定每个维度上的步幅大小。

  • padding (字符串): 可选参数，指定填充方式。支持取值为 'valid'（不填充）或 'same'（填充），默认值为 'valid'。

返回值

一个 tf.layers.Layer 实例，代表创建的最大池化层。

示例

const model = tf.sequential();
model.add(tf.layers.maxPooling3d({poolSize: 2, strides: 2, padding: 'valid', inputShape: [32, 32, 32, 3]}));

// 可以将多个最大池化层堆叠起来
model.add(tf.layers.maxPooling3d({poolSize: [2, 2, 2], strides: [1, 1, 1], padding: 'valid'}));
model.add(tf.layers.maxPooling3d({poolSize: 3}));

引用

• TensorFlow.js API 文档: tf.layers.maxPooling3d()

• TensorFlow.js Python API 文档: tf.keras.layers.MaxPooling3D