tf.keras.layers.ZeroPadding2D

tf.keras.layers.ZeroPadding2D은 2D 입력 (e.g. 이미지)에 대해 Zero-padding layer를 반환합니다.

예제

import tensorflow as tf
import numpy as np


input_shape = (1, 1, 2, 2)
x = np.prod(input_shape)
print(x)

y = np.arange(x).reshape(input_shape)
print(y)

4
[[[[0 1]
   [2 3]]]]

numpy.prod 함수는 어레이 요소의 곱을 반환합니다.

numpy.arange, numpy.reshape 함수를 사용해서 (1, 1, 2, 2) 형태의 어레이를 만들었습니다.

z = tf.keras.layers.ZeroPadding2D(padding=1)(y)
print(z)

tf.Tensor(
[[[[0 0]
   [0 0]
   [0 0]
   [0 0]]

[[0 0]
 [0 1]
 [2 3]
 [0 0]]

[[0 0]
 [0 0]
 [0 0]
 [0 0]]]], shape=(1, 3, 4, 2), dtype=int64)

tf.keras.layers.ZeroPadding2D(padding=1)(y)을 사용해서 어레이 y의 주변에 0을 덧붙였습니다.

이전글/다음글

이전글 : tf.keras.layers.Minimum

다음글 : tf.keras.metrics.Accuracy

공유하기