Tensorflowtf.argmax函数
Posted xqnq2007
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Tensorflowtf.argmax函数相关的知识,希望对你有一定的参考价值。
tf.argmax(input, axis=None, name=None, dimension=None)
此函数是对矩阵按行或列计算最大值
参数
- input:输入Tensor
- axis:0表示按列,1表示按行
- name:名称
- dimension:和axis功能一样,默认axis取值优先。新加的字段
例:
以上是关于Tensorflowtf.argmax函数的主要内容,如果未能解决你的问题,请参考以下文章
在 Visual Studio 中创建构造函数的代码片段或快捷方式