reduce 可以理解为 python 里的 reduce 函数;
tensorflow 中有很多 reduce_ API,其用法完全相同
tf.reduce_max
以这个为例进行说明
def reduce_max(input_tensor, axis=None, keepdims=None, name=None, reduction_indices=None, keep_dims=None): """Computes the maximum of elements across dimensions of a tensor.
求指定维度上的最大值
input_tensor:输入数据,tensor、array、dataframe 都可以
axis:维度,0 表示 列,1 表示行,-1 表示最后一个维度,可以是标量和列表
keepdims:bool 值,按原维度输出
reduction_indices:等价于 axis,已废弃
示例
data = tf.constant([[1, 2], [3, 4], [2, 5]]) print(data.shape) # (3, 2) out1 = tf.reduce_max(data, axis=0) ### 第 0 个维度即 每列最大值 out2 = tf.reduce_max(data, axis=1) ### 第 1 个维度即 每行最大值 out3 = tf.reduce_max(data, axis=0, keepdims=True) ### 保持原维度 out4 = tf.reduce_max(data, reduction_indices=[1]) ### 等价于 axis=1 out5 = tf.reduce_max(data, axis=[0, 1]) ### 多个维度最大值 sess = tf.Session() print(sess.run(out1)) # [3 5] print(sess.run(out2)) # [2 4 5] print(sess.run(out3)) # [[3 5]] print(sess.run(out4)) # [2 4 5] print(sess.run(out5)) # 5
reduce_
tf.reduce_min:最小
tf.reduce_sum:求和
tf.reduce_mean:均值
tf.reduce_all:计算张量在维度上的逻辑和
tf.reduce_any:在张量的维度上计算元素的 逻辑或
参考资料:
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 记一次.NET内存居高不下排查解决与启示
· 探究高空视频全景AR技术的实现原理
· 理解Rust引用及其生命周期标识(上)
· 浏览器原生「磁吸」效果!Anchor Positioning 锚点定位神器解析
· 没有源码,如何修改代码逻辑?
· 全程不用写代码,我用AI程序员写了一个飞机大战
· DeepSeek 开源周回顾「GitHub 热点速览」
· 记一次.NET内存居高不下排查解决与启示
· MongoDB 8.0这个新功能碉堡了,比商业数据库还牛
· .NET10 - 预览版1新功能体验(一)
2019-02-28 cnn-手写数字识别
2019-02-28 基础理论-极大似然
2019-02-28 决策树-缺失值处理