Python - tensorflow.math.subtract()

简介

tensorflow.math.subtract()函数是TensorFlow中的数学计算函数之一，用于对两个张量中的元素进行减法运算。如果维度不同，则会自动进行广播(broadcast)操作。

语法

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

参数说明

• x: 张量，数据类型必须是float16、float32、float64、int32、complex64、complex128或bool。
• y: 张量，数据类型必须和x相同。
• name: （可选）操作的名称。

返回值

张量，数据类型和形状与输入(x、y)相同。

示例

import tensorflow as tf

a = tf.constant([[1, 2], [3, 4]])
b = tf.constant(1)
c = tf.math.subtract(a, b)
print(c.numpy())

输出结果为：

[[0 1]
 [2 3]]

注意事项

• 输入的x、y张量必须形状相同或能够进行广播操作，否则会抛出异常。
• 支持复数运算，但要求数据类型为complex64和complex128。