tensorflow add、subtarct函数

tf.add(
    x, y, name=None
)

x和y两个张量进行元素级的相加。如果x和y的形状不一致，则会利用python的广播机制，扩展x或y的形状，使两个张量的形状相同，然后再进行相加。

1、x和y的形状相同

>>> x = tf.ones((2, 2))
>>> x

>>> y = tf.reshape(tf.range(4), (2, 2))
>>> y = tf.cast(y, dtype=x.dtype)
>>> y

>>> tf.add(x, y)

2、x和y的形状不一致

>>> x = tf.ones((2, 2))
>>> x

>>> y = tf.constant([1, 1], dtype=x.dtype)
>>> tf.add(x, y)

>>> y = tf.constant([[1], [1]], dtype=x.dtype)
>>> tf.add(x, y)

>>> y = tf.constant([1], dtype=x.dtype)
>>> tf.add(x, y)

tf.subtract(
    x, y, name=None
)

张量x和张量y实现元素级的相减。如果x和y的形状不一样，也会使用python的广播机制。

1、x和y形状一致

>>> x = tf.fill((2, 2), 3)
>>> x

>>> y = tf.reshape(tf.range(4), (2, 2))
>>> y

>>> tf.subtract(x, y)

2、x和y形状不一致

>>> x = tf.fill((2, 2), 3)
>>> x

>>> y = tf.constant([[1, 1]])
>>> y

>>> tf.subtract(x, y)