tf.argmax() axis解释
axis最简单的解释就是返回最大的那个数值的下标,但是新手不好理解。
下面详细解释:
现在有一个张量
a = [1,2,3],是个一维张量
tf.argmax(a,0) 返回值为2,此时表示返回张量a第一个维度中最大元素的下标,a只有一维,最大元素为3,其下标为2(下标从0开始数)
tf.argmax(a,1) 这个会报错,因为a是一维张量,axis=0代表第一维度,axis=1代表第二...
原创
2019-09-28 19:26:52 ·
1422 阅读 ·
3 评论