1. 程式人生 > >TensorFlow版本帶來的concat錯誤

TensorFlow版本帶來的concat錯誤

錯誤提示:

TypeError: Expected int32, got list containing Tensors of type ‘_Message’ instead.

錯誤說明:

根據提示知道程式碼中一行concat相關的程式碼。
是由於TensorFlow版本帶來的錯誤。

在TensorFlow 1.0以前的版本(0.x)的API中,concat的引數是數字在前,tensors在後的:

tf.concat(3, net, name=name)

而在TensorFlow 1.0版本以後的API中,concat的引數是tensors在前,數字在後的:

tf.concat(net, 3
, name=name)

因為參考的程式碼可能當時執行的TensorFlow版本與本機版本不同,所以有了問題。

解決方案:

根據錯誤提示找到對應程式碼行,把concat的引數調換一下順序就可以成功運行了。