Класс Subtract
Класс Subtract выполняет поэлементное вычитание
входных тензоров. Слой принимает список тензоров
одинаковой формы и вычитает из первого тензора все
последующие. Первым параметром передается список
входных тензоров, вторым - дополнительные именованные
аргументы, такие как name и dtype.
Слой удобно применять в функциональном API Keras
для построения остаточных связей.
Синтаксис
tf.keras.layers.Subtract(**kwargs)
Пример
Давайте создадим слой Subtract и вычтем
из первого тензора второй:
import tensorflow as tf
layer = tf.keras.layers.Subtract()
t1 = tf.constant([1, 2, 3, 4, 5])
t2 = tf.constant([1, 1, 1, 1, 1])
res = layer([t1, t2])
print(res)
Результат выполнения кода:
tf.Tensor([0 1 2 3 4], shape=(5,), dtype=int32)
Пример
Давайте применим слой Subtract к двумерным
тензорам:
import tensorflow as tf
layer = tf.keras.layers.Subtract()
t1 = tf.constant([[1, 2, 3], [4, 5, 6]])
t2 = tf.constant([[1, 1, 1], [2, 2, 2]])
res = layer([t1, t2])
print(res)
Результат выполнения кода:
tf.Tensor(
[[0 1 2]
[2 3 4]], shape=(2, 3), dtype=int32)
Пример
Давайте вычтем три тензора сразу, слой вычтет второй и третий тензоры из первого:
import tensorflow as tf
layer = tf.keras.layers.Subtract()
t1 = tf.constant([1, 2, 3, 4, 5])
t2 = tf.constant([1, 1, 1, 1, 1])
t3 = tf.constant([0, 1, 2, 3, 4])
res = layer([t1, t2, t3])
print(res)
Результат выполнения кода:
tf.Tensor([0 0 0 0 0], shape=(5,), dtype=int32)