摘要: tf.argmax(input,axis)根据axis取值的不同返回每行或者每列最大值的索引。代码如下: import tensorflow as tfimport numpy as npsess=tf.Session()a = np.array([[1, 2, 3], [2, 3, 4], [5, 阅读全文
posted @ 2019-12-09 23:30 tangjunjun 阅读(485) 评论(0) 推荐(0) 编辑
https://rpc.cnblogs.com/metaweblog/tangjunjun