这个函数有两个重要参数：
tf.argmax(x，维度）
其中x是一个矩阵，维度可以0，1，2等；
像tf.argmax(x,0)就是在行里里面比较大小，并返回最大值的下标，
1：则为按列比较大小，并返回最大值的下标
2：在第3维度比较大小，并返回该维度最大值的下标
tf.argmax
猜你喜欢
转载自blog.csdn.net/weixin_38859557/article/details/80878333
今日推荐
周排行