tf.keras.layers.ZeroPadding2D

View source on GitHub

Zero-padding layer for 2D input (e.g. picture).

Inherits From: Layer

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

This layer can add rows and columns of zeros at the top, bottom, left and right side of an image tensor.

Arguments:

Input shape:

4D tensor with shape: - If data_format is "channels_last": (batch, rows, cols, channels) - If data_format is "channels_first": (batch, channels, rows, cols)

Output shape:

4D tensor with shape: - If data_format is "channels_last": (batch, padded_rows, padded_cols, channels) - If data_format is "channels_first": (batch, channels, padded_rows, padded_cols)