tf.argmax()以及axis

tf.argmax()表示返回最大值的索引号，axis为0 ，表示返回每列最大值索引号。axis为1 ，表示返回每行最大值索引号

结果为

posted @ 2019-03-10 21:42  冥然兀坐，万籁有声  阅读(287)  评论(0)  编辑  收藏  举报