Tensorflow 2 張量

Question

目前在使用 Tensorflow 2 時我的張量出現以下錯誤：

tensorflow:Model 是用形狀 (None, 8) 構建的，用於輸入 KerasTensor(type_spec=TensorSpec(shape=(None, 8), dtype=tf.float32, name='dense_input'), name='dense_input', description="由層 'dense_input'" 創建），但它在形狀不兼容的輸入上被調用（無、1、8）。

只是想知道我如何重塑它以獲得 None,8 的形狀而不是 None, 1, 8。我之前使用過 tf.reshape 但我不確定在這種情況下如何使用它。

Answer 1

https://www.tensorflow.org/api_docs/python/tf/squeeze

tf.squeeze(input)將刪除所有大小為 1 的軸

或者

tf.squeeze(input, 1)將tf.squeeze(input, 1)刪除第一個軸