TensorFlow学习笔记--（3）张量的常用运算函数

损失函数及求偏导

通过 tf.GradientTape 函数来指定损失函数的变量以及表达式

最后通过 gradient(%损失函数%,%偏导对象%) 来获取求偏导的结果

独热编码

给出一组特征值 来对图像进行分类 可以用独热编码

0的概率是第0种 1的概率是第1种 0的概率是第二种

tf.one_hot(%某标签值%,%分类数%)
这里还没太看懂结果的3X3矩阵是怎么来的 如果单纯的是因为有几种类型 就有几个1
那传入的标签值参数就没有意义了
1
2
3

张量转换输出概率分布

tf.nn.softmax(%张量%)
传入一组张量值 根据每个分量的大小 来输出每种分类的概率值
1
2

返回指定维度最大值所在的索引

对axis=0 求每一列的最大值所在的索引号 分别是 3 3 1

对axis=1 则是求每一行的