zoukankan      html  css  js  c++  java
  • TensorFlow函数(七)tf.argmax()

    tf.argmax(input, dimension, name=None)

    参数:

    • input:输入数据
    • dimension:按某维度查找。

        dimension=0:按列查找;

        dimension=1:按行查找;

    返回:

    • 最大值的下标
    1 a = tf.constant([1.,2.,3.,0.,9.,])
    2 b = tf.constant([[1,2,3],[3,2,1],[4,5,6],[6,5,4]])
    3 with tf.Session() as sess:
    4     sess.run(tf.argmax(a, 0))
    5 with tf.Session() as sess:
    6     sess.run(tf.argmax(b, 0))
    7 with tf.Session() as sess:
    8     sess.run(tf.argmax(b, 1))

    输出:

    4

    输出:

    [3, 2, 2]

    输出:

    [2, 0 ,2, 0]

  • 相关阅读:
    博客园如何运行代码
    视觉差
    h5 播放器 -3
    播放器 视频 音频 -1
    游戏 保卫萝卜
    跟踪算法
    走口字

    联动日历
    jq 抽奖
  • 原文地址:https://www.cnblogs.com/reaptomorrow-flydream/p/9490383.html
Copyright © 2011-2022 走看看