当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python tf.keras.layers.ZeroPadding3D用法及代码示例


3D 数据的零填充层(空间或spatio-temporal)。

继承自:LayerModule

用法

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

参数

  • padding Int,或 3 个 int 的元组,或 3 个 2 个 int 的元组的元组。
    • 如果 int: 相同的对称填充应用于高度和宽度。
    • 如果 3 个整数的元组:解释为高度和宽度的两个不同的对称填充值:(symmetric_dim1_pad, symmetric_dim2_pad, symmetric_dim3_pad).
    • If tuple of 3 tuples of 2 ints: 解释为((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) 的输入。它默认为您的 Keras 配置文件中的 image_data_format~/.keras/keras.json 。如果您从未设置它,那么它将是"channels_last"。

例子:

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)

输入形状:

具有形状的 5D 张量:

  • 如果 data_format"channels_last"(batch_size, first_axis_to_pad, second_axis_to_pad, third_axis_to_pad, depth)
  • 如果 data_format"channels_first"(batch_size, depth, first_axis_to_pad, second_axis_to_pad, third_axis_to_pad)

输出形状:

具有形状的 5D 张量:

  • 如果 data_format"channels_last"(batch_size, first_padded_axis, second_padded_axis, third_axis_to_pad, depth)
  • 如果 data_format"channels_first"(batch_size, depth, first_padded_axis, second_padded_axis, third_axis_to_pad)

相关用法


注:本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.keras.layers.ZeroPadding3D。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。