在上一篇文章中我们成功的输入了数据,下一步就是利用这些数据进行计算,本文将介绍最基础的集中计算方式–加减乘除。更多资料请查看
官方文档。
加法
x = tf.add(5, 2) # 7
减法
x = tf.subtract(10, 4) # 6
乘法
y = tf.multiply(2, 5) # 10
除法
y = tf.div(10, 5) # 2
类型转换
当你尝试计算两个不同类型的值的时候会出现报错,例如:
tf.subtract(tf.constant(2.0),tf.constant(1))
""" ValueError: Tensor conversion requested dtype float32 for Tensor with dtype int32: """
此时就需要使用tf.cast()转换类型:
tf.subtract(tf.cast(tf.constant(2.0), tf.int32), tf.constant(1)) # 1