tensorflow学习
在跑minist demo时,遇到了这几句
BATCHSIZE=6
label=tf.expand_dims(tf.constant([0,2,3,6,7,9]),1)
index=tf.expand_dims(tf.range(0,BATCHSIZE),1)
concated = tf.concat(1, [index, label])
可以跑的时候却报错
查了半天资料,发现大多数都是这么写。。应该没错啊。。。
最后发现,在tensorflow 1.3.0 以后,concated的API 变了,卧槽。。。神坑啊
正确的写法是
concated = tf.concat([index, label], 1)
参数的顺序竟然换了,尼玛。。。
结果正确显示为: