小贝子编程

TensorFlow张量中最大元素的索引



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

对不起，我已经想好了。它是：

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

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

相关内容

  • 没有找到相关文章

最新更新


热门标签：

javascript python java c# php android html jquery c++ css ios sql mysql arrays asp.net json python-3.x ruby-on-rails .net sql-server django objective-c excel regex ruby linux ajax iphone xml vba spring asp.net-mvc database wordpress string postgresql wpf windows xcode bash git oracle list vb.net multithreading eclipse algorithm macos powershell visual-studio image forms numpy scala function api selenium