「tensorflow坑点记录」No.1:不同版本的tf.concat函数报错

错误语句:h3_concat = tf.concat( 3 , [h3, tiled_embeddings], name='h3_concat')

报错信息:TypeError: Expected int32, got list containing Tensors of type '_Message' instead.

原因:

tensorflow版本问题。1.0版本以前这样写是对的,但是1.0及以后,函数的原型为

tf.stack(tensors, axis=axis)

即数字在前,tensors在后。

原代码修改为

h3_concat = tf.concat([h3, tiled_embeddings],  3 , name='h3_concat')

即可√

发表评论

电子邮件地址不会被公开。