tf.argmax
TF
2023-09-14 09:15:50 时间
tf.argmax(input, axis=None, name=None, dimension=None)
此函数是对矩阵按行或列计算最大值
参数
input:输入Tensor
axis:0表示按列,1表示按行
name:名称
dimension:和axis功能一样,默认axis取值优先。新加的字段
返回:Tensor 一般是行或列的最大值下标
import tensorflow as tf
a=tf.Variable(tf.random_uniform([3,4],minval=-1,maxval=1))
b=tf.argmax(input=a,axis=0)
c=tf.argmax(input=a,dimension=1) #此处用dimesion或用axis是一样的
sess = tf.Session()
sess.run(tf.initialize_all_variables())
print(sess.run(a))
#[[ 0.04261756 -0.34297419 -0.87816691 -0.15430689]
# [ 0.18663144 0.86972666 -0.06103253 0.38307118]
# [ 0.84588599 -0.45432305 -0.39736366 0.38526249]]
print(sess.run(b))
#[2 1 1 2]
print(sess.run(c))
#[0 1 0]
相关文章
- SD卡、TF卡、MMC卡、emmc、sdio扫盲
- TensorFlow 2.10上线:Windows上扩展GPU支持,TF-DF 1.0发布
- tf.placeholder函数说明
- 汇编语言中的标志位:CF、PF、AF、ZF、SF、TF、IF、DF、OF
- EEG时频主成分分析(TF-PCA)实用教程(附示例数据和代码)
- 【说站】Python中Tf-idf文本特征的提取
- Hadoop代做编程辅导:CA675 TF-IDF
- tf.to_float
- R语言文本挖掘NASA数据网络分析,tf-idf和主题建模|附代码数据
- 高级API、异构图:谷歌发布TF-GNN,在TensorFlow中创建图神经网络
- 2023年TF活动正式启动
- 德商必酷推出Dark Rock TF 2散热器 可压制230W处理器