随笔 - 1762  文章 - 0  评论 - 109  阅读 - 431万

tf.reduce_max()

作用:找到某个维度的最大值。

tf.reduce_max()函数
tf.reduce_max(
input_tensor,
axis=None,
name=None,
keepdims=False #是否保持矩形原狀

参数解释:
input_tensor:输入数据,tensor、array、dataframe 都可以

axis:表示维度,从0开始表示最外层维度,也可以用-1表示最内层维度;

[0, [1, [2, [3,[...]]]]],或者[[[[[...], -4], -3], -2], -1] 数字表示对应[ ]的维度。当axis位默认表示求全局最大值。

keepdims:默认时,维度会减少1,为True时,保持维度不变。

name:操作名称

reduction_indices:axis的旧名称(已经弃用)

示例:
import tensorflow as tf
import numpy as np

a=np.array([[1, 2],
[5, 3],
[2, 6]])

b = tf.Variable(a)
with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
print(sess.run(b))
print('************')
# 对于二维矩阵,axis=0轴可以理解为行增长方向(向下)即按列求最值,axis=1轴可以理解为列增长方向(向右)按列行求最值
print(sess.run(tf.reduce_max(b, axis=1, keepdims=False))) # keepdims=False,axis=1被消减,不保持原状,本来shape为(3,1),后来变成(1,3)了
print('************')
print(sess.run(tf.reduce_max(b, axis=1, keepdims=True)))
print('************')
print(sess.run(tf.reduce_max(b, axis=0, keepdims=True)))
结果:

[[1 2]
[5 3]
[2 6]]
************
[2 5 6]
************
[[2]
[5]
[6]]
************
[[5 6]]
ps:另外该函数等价于np.max:(a, axis=None, out=None, keepdims=False)

import tensorflow as tf
import numpy as np

max_value = tf.reduce_max([[1, 3, 2], [4,5,6]], axis=0)
with tf.Session() as sess:
max_value = sess.run(max_value)
print(max_value)

print(np.max([[1, 3, 2], [4,5,6]], axis=0))


#### 输出
[4 5 6]
[4 5 6]
————————————————
版权声明:本文为CSDN博主「无尽的沉默」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/hgnuxc_1993/article/details/116940059

posted on   一杯明月  阅读(449)  评论(0编辑  收藏  举报
编辑推荐:
· AI与.NET技术实操系列:向量存储与相似性搜索在 .NET 中的实现
· 基于Microsoft.Extensions.AI核心库实现RAG应用
· Linux系列:如何用heaptrack跟踪.NET程序的非托管内存泄露
· 开发者必知的日志记录最佳实践
· SQL Server 2025 AI相关能力初探
阅读排行:
· winform 绘制太阳,地球,月球 运作规律
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· AI与.NET技术实操系列(五):向量存储与相似性搜索在 .NET 中的实现
· 超详细:普通电脑也行Windows部署deepseek R1训练数据并当服务器共享给他人
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
历史上的今天:
2020-08-18 matlab fullfile()函数
< 2025年3月 >
23 24 25 26 27 28 1
2 3 4 5 6 7 8
9 10 11 12 13 14 15
16 17 18 19 20 21 22
23 24 25 26 27 28 29
30 31 1 2 3 4 5

点击右上角即可分享
微信分享提示