TensorFlow笔记 常用函数

一、tf.cast
强制数据类型转换

二、tf.reduce_min
计算最小值

三、tf.reduce_max
计算最大值

四、tf.reduce_mean
求平均值

五、tf.reduce_sum
求和

六、tf.Variable
标记为变量

七、数学函数
1.四则运算：tf.add、tf.subtract、tf.multiply、tf.divide

2.平方、次方、开方：tf.square、tf.pow、tf.sqrt

3.矩阵乘：tf.matmul

八、tf.data.Dataset.from_tensor_slices
将特征与标签配对

九、tf.GradientTape
对变量求梯度

十、enumerate（python的内建函数）
遍历每个元素

十一、tf.one_hot
独热编码（分类问题）

十二、tf.nn.softmax
使输出符合概率分布（和为1.0）

十三、变量.assign_sub
自减

十四、变量.assign_add
自增

十五、tf.argmax
找出最大值的索引

十六、tf.argmin
找出最小值的索引