tf.argmax()中的axis参数

来源:互联网 发布:word是网络通信软件吗 编辑:程序博客网 时间:2024/06/05 13:28
tf.argmax()中的axis在等于0时,按列比较,取得最大值的下标索引;等于1时,按行比较。