tf.keras.layers.Maximum

tf.keras.layers.Maximum은 입력값 중 최대값을 계산합니다.



예제1

import tensorflow as tf
import numpy as np

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

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

print(y)

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

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

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



예제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.Maximum()([x1, x2])
print(y.shape)

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

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

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



이전글/다음글


공유하기