tensorflow学习之常用函数:tf.reduce

发布时间:2024-12-03 07:38

深度学习中常用的激活函数解析 #生活技巧# #学习技巧# #深度学习技巧#

最新推荐文章于 2020-07-09 10:25:10 发布

搞视觉的张小凡 于 2018-07-22 21:03:12 发布

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。

前沿

学习TensorFlow的时候,经常要编写损失函数的代码,经常需要用到tf.reduce_mean,tf.reduce_sum等函数。

reduce_max

求最大值tf.reduce_max(input_tensor, reduction_indices=None, keep_dims=False, name=None)

参数1--input_tensor:待求值的tensor。

参数2--reduction_indices:在哪一维上求解。表示函数的处理维

若reduction_indices=[0]:  表示在对行进行操作,将行进行压缩。将每一行全部进行biji比较码如下:

若reduction_indices=[1]:表示对列进行操作,将列进行压缩。将每一列全部进行比较,具体代码如下:

import tensorflow as tf

arr = tf.constant([1, 2, 3, 4, 5, 6],shape=[2, 3])

sess=tf.Session()

print(sess.run(arr))

print(sess.run(tf.reduce_mean(arr,reduction_indices=[0])))

arr:[[1 2 3]
 [4 5 6]]

若reduction_indices=[0]:结果为[4 5 6]

若reduction_indices=[1]:结果为[3  6] 

      需要注意的一点,在很多的时候,我们看到别人的代码中并没有reduction_indices这个参数,此时该参数取默认值None,将把input_tensor降到0维,也就是一个数                                          

参数(3)(4)可忽略

reduce_sum

求平均值tf.reduce_mean(input_tensor, reduction_indices=None, keep_dims=False, name=None)

参数同上:

网址:tensorflow学习之常用函数:tf.reduce https://www.yuejiaxmz.com/news/view/356851

相关内容

Tensorflow笔记之【神经网络的优化】
batch normalization tensorflow实现的正确姿势
np.prod() 函数计算数组元素乘积等
机器学习Machine Learning:成本(cost) 函数,损失(loss)函数,目标(Objective)函数的区别和联系?
tensorflow 1.9.0 语音识别简单实现
链接、资源(学习)
高三数学一轮复习精品课件2:函数y=Asin(ωx+φ)的图象及三角函数模型的应用.ppt
Kotlin之扩展函数
一次函数在生活中的应用--
python函数练习

随便看看