刻苦皮卡丘

文章
5
资源
0
加入时间
2年10月21天

tf.argmax()以及axis解析

用tensorflow做CNN_TEXT文本分类时,看到这个API,然后去官网查了一下,再看了一下别的资料,算是明白它的处理方式了。首先,明确一点,tf.argmax可以认为就是np.argmax。  简单的说,tf.argmax就是返回最大的那个数值所在的下标。这个很好理解,只是tf.argmax()的参数让人有些迷惑,比如,tf.argmax(array, 1)和tf.argmax(arr