當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


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。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。