TensorFlow中张量减法操作tf.subtract用法

一、环境

TensorFlow API r1.12

CUDA 9.2 V9.2.148

cudnn64_7.dll

Python 3.6.3

Windows 10

二、官方说明

张量x减去张量y（元素级减法操作）

https://www.tensorflow.org/api_docs/python/tf/math/subtract

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

注意：很多旧版本的TensorFlow代码中的tf.sub()已经废弃，目前的张量减法操作名称是：tf.math.subtract 或 tf.subtract

 

输入：

（1）x：被减张量，支持的数据类型 bfloat16, half, float32, float64, uint8, int8, uint16, int16, int32, int64, complex64, complex128

（2）y：减数张量，支持的数据类型同上

（3）name：可选参数，操作的名称

返回结果：

（1）与被减数张量具有相同类型的张量

 

三、实例

>>> import tensorflow as tf
>>> sess = tf.InteractiveSession()
>>> v = tf.Variable([1.0,2.0])
>>> c = tf.constant([3.0,4.0])
>>> v.initializer.run()
>>> sub = tf.subtract(v,c)
>>> print(sub.eval())
[-2. -2.]