Keras 2 API 文档 / 层 API / 形状变换层 / ZeroPadding3D 层

ZeroPadding3D 层

[源代码]

ZeroPadding3D

tf_keras.layers.ZeroPadding3D(padding=(1, 1, 1), data_format=None, **kwargs)

用于 3D 数据(空间数据或时空数据)的零填充层。

示例

>>> input_shape = (1, 1, 2, 2, 3)
>>> x = np.arange(np.prod(input_shape)).reshape(input_shape)
>>> y = tf.keras.layers.ZeroPadding3D(padding=2)(x)
>>> print(y.shape)
(1, 5, 6, 6, 3)

参数

  • padding: 整数,或包含 3 个整数的元组,或包含 3 个元组(每个元组包含 2 个整数)的元组。
    • 如果是整数:将对高度和宽度应用相同的对称填充。
    • 如果是包含 3 个整数的元组:被解释为对三个维度分别应用的对称填充值:(symmetric_dim1_pad, symmetric_dim2_pad, symmetric_dim3_pad)
    • 如果是包含 3 个元组(每个元组包含 2 个整数)的元组:被解释为 ((left_dim1_pad, right_dim1_pad), (left_dim2_pad, right_dim2_pad), (left_dim3_pad, right_dim3_pad))
  • data_format: 字符串,以下值之一:channels_last(默认)或 channels_first。输入中维度的顺序。channels_last 对应形状为 (batch_size, spatial_dim1, spatial_dim2, spatial_dim3, channels) 的输入,而 channels_first 对应形状为 (batch_size, channels, spatial_dim1, spatial_dim2, spatial_dim3) 的输入。未指定时,使用 TF-Keras 配置文件 ~/.keras/keras.json 中找到的 image_data_format 值(如果存在),否则使用 'channels_last'。默认为 'channels_last'。

输入形状

5D 张量,形状为: - 如果 data_format"channels_last"(batch_size, 待填充的第一个轴, 待填充的第二个轴, 待填充的第三个轴, 深度) - 如果 data_format"channels_first"(batch_size, 深度, 待填充的第一个轴, 待填充的第二个轴, 待填充的第三个轴)

输出形状

5D 张量,形状为: - 如果 data_format"channels_last"(batch_size, 填充后的第一个轴, 填充后的第二个轴, 待填充的第三个轴, 深度) - 如果 data_format"channels_first"(batch_size, 深度, 填充后的第一个轴, 填充后的第二个轴, 待填充的第三个轴)