tf.argmax()

tf.argmax() 返回矩阵横列或者纵列的最大值的坐标,取决于第二个参数

参数:

tf.argmax(
    input,
    axis=None,//0是纵列 1是横列
    name=None,
    dimension=None,
    output_type=tf.int64
)

使用案例:

import tensorflow as tf

x = tf.constant([[0, 2, 1], [2, 1, 3]])
sess = tf.Session()
print(sess.run(tf.argmax(x, 0)))//[1 0 1]
print(sess.run(tf.argmax(x, 1)))//[1 2]

tf.argmax(vector, 1):返回的是vector中的最大值的索引号,如果vector是一个向量,那就返回一个值,如果是一个矩阵,那就返回一个向量,这个向量的每一个维度都是相对应矩阵行的最大值元素的索引号。

import tensorflow as tf
import numpy as np
 
A = [[1,3,4,5,6]]
B = [[1,3,4], [2,4,1]]
 
with tf.Session() as sess:
    print(sess.run(tf.argmax(A, 1)))
    print(sess.run(tf.argmax(B, 1)))

输出:

[4]
[2 1]

参考:https://blog.csdn.net/uestc_c2_403/article/details/72232807

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • tf.InteractiveSession()与tf.Session()

    tf.InteractiveSession():它能让你在运行图的时候,插入一些计算图,这些计算图是由某些操作(operations)构成的。这对于工作在交互式...

    周小董
  • tf.cast()

    参考:https://www.cnblogs.com/hezhiyao/p/8196587.html

    周小董
  • Tensorflow模型保存和读取tf.train.Saver

    然后,在训练循环中,定期调用 saver.save() 方法,向文件夹中写入包含了当前模型中所有可训练变量的 checkpoint 文件。

    周小董
  • tensorflow(一)windows 10 64位安装tensorflow1.4与基本概念解读tf.global_variables_initializer

    一.安装 目前用了tensorflow、deeplearning4j两个深度学习框架, tensorflow 之前一直支持到python 3.5,目前以更新...

    学到老
  • tensorflow(一)windows 10 64位安装tensorflow1.4与基本概念解读tf.global_variables_initializer

    目前用了tensorflow、deeplearning4j两个深度学习框架, tensorflow 之前一直支持到python 3.5,目前以更新到3.6,...

    学到老
  • 自学tensorflow——2.使用te

    2.写一个函数generate_data(),用来生成我们所需要的数据,这里使用的线性函数是y = 0.1*x + 0.3,具体解释见注释

    py3study
  • 深度学习(1)——tensorflow简介什么是TensorFlow?什么是数据流图?安装基本概念示例变量的更新操作

    DC童生
  • 人工智能python的tensorflow基础

    IT故事会
  • TensorFlow线性回归与逻辑回归实战

    Huber loss是为了增强平方误差损失函数(squared loss function)对噪声(或叫离群点,outliers)的鲁棒性提出的。

    公众号guangcity
  • 使用Python实现touch命令功能

           公司指定的办公平台是Windows,而且给了诸多不自由的限制。如果对shell有一点点依赖,那么会是一个很麻烦的事情,毕竟对Windows的批处理...

    py3study

扫码关注云+社区

领取腾讯云代金券