摘要: tf.argmax(input, axis=None, name=None, dimension=None) axis: 0 表示按列,1 表示按行 tf.boolean_mask(a,b) 其中 b 一般是 bool 型的 n 维向量,若 a.shape=[3,3,3], b.shape=[3,3 阅读全文
posted @ 2018-04-18 15:07 推杯问盏 阅读(298) 评论(0) 推荐(0) 编辑