[TensorFlow入门] 数学计算

在上一篇文章中我们成功的输入了数据,下一步就是利用这些数据进行计算,本文将介绍最基础的集中计算方式–加减乘除。更多资料请查看
官方文档

加法

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
    原文作者:MAZE
    原文地址: https://zhuanlan.zhihu.com/p/25311333
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞