gpt4 book ai didi

linear-algebra - TensorFlow 张量中最大元素的索引

转载 作者:行者123 更新时间:2023-12-04 16:51:34 25 4
gpt4 key购买 nike

如何在 TensorFlow 中沿选定轴获取张量的最大元素的索引?

最佳答案

抱歉,已经想通了。这是:

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

https://www.tensorflow.org/api_docs/python/tf/argmax

关于linear-algebra - TensorFlow 张量中最大元素的索引,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/37104287/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com