tf.keras.layers.Minimum

tf.keras.layers.Minumum은 입력값의 최소값을 계산합니다.



예제1

import tensorflow as tf
import numpy as np

## Ex01
x1 = np.arange(5).reshape(5, 1)
x2 = np.arange(5, 10).reshape(5, 1)
print(x1)
print(x2)

y = tf.keras.layers.Minimum()([x1, x2])

print(y)

[[0]
[1]
[2]
[3]
[4]]
[[5]
[6]
[7]
[8]
[9]]
tf.Tensor(
[[0]
[1]
[2]
[3]
[4]], shape=(5, 1), dtype=int64)

tf.keras.layers.Minimum은 동일한 형태를 갖는 텐서를 입력으로 받고,

각 텐서 값들의 최소값을 갖는 동일한 형태의 텐서를 반환합니다.



예제2

import tensorflow as tf
import numpy as np

x1 = tf.keras.layers.Dense(8)(np.arange(6).reshape(3, 2))
x2 = tf.keras.layers.Dense(8)(np.arange(6, 12).reshape(3, 2))
print(x1.shape)
print(x2.shape)

y = tf.keras.layers.Minimum()([x1, x2])
print(y.shape)

(3, 8)
(3, 8)
(3, 8)

x1, x2는 각각 tf.keras.layers.Dense() 레이어의 출력으로서 (3, 8) 형태를 갖는 텐서입니다.

y는 tf.keras.layers.Minimum()([x1, x2])의 출력으로서 입력 텐서들과 동일한 형태를 갖습니다.



이전글/다음글


공유하기