tensor 最大值的下标的求法

最新推荐文章于 2023-07-14 23:48:13 发布

原创最新推荐文章于 2023-07-14 23:48:13 发布 · 9k 阅读

1 ·

CC 4.0 BY-SA版权

python 专栏收录该内容

7 篇文章

订阅专栏

本文通过使用TensorFlow和NumPy进行数据操作的实例，展示了如何寻找数组中的最大值及其索引，包括将TensorFlow张量转换为NumPy数组并应用NumPy函数的过程。

部署运行你感兴趣的模型镜像

import tensorflow as tf
import numpy as np

a=tf.constant([1,0,-1,12,5],dtype=tf.float32)
# b=tf.argmax(input(a),axis=1) 出错
flag=0

with tf.Session() as sess:
    print(sess.run(a))
    b = a.eval()

    for i in range(4):
        c=b[i]
        if c>=b[i]:
            flag=i
    print(flag)


    b = tf.constant([1, 0, -1, 12, 5], dtype=tf.float32)
    b=b.eval().tolist() #b转为list，调用函数index(max())求最大值下标
    dd = b.index(max(b))
    print(dd)

ar=np.array([[1,2,3,4],[5,6,7,8]])
print(ar)
print(np.argmax(ar,axis=1))

您可能感兴趣的与本文相关的镜像