Sequential modeの中でミニバッチサイズを変更したい

前提・実現したいこと

tensorflow2.0を使用して、Sequential モデルを作成しています。
Sequential モデルの中で、reshapeを使用してミニバッチサイズを変更したいです。

具体的には、ミニバッチサイズが32のとき、下記のようにreshapeしたいです。
(32, 10, 8, 8, 1)　→　(32*10, 8, 8, 1)
つまり、(?, 10, 8, 8, 1)　→　(?, 8, 8, 1)のreshapeを行いたい状況です。

発生している問題・エラーメッセージ

InvalidArgumentError:  Input to reshape is a tensor with 20480 values, but the requested shape has 2048

該当のソースコード

Python
1model.add(tf.keras.layers.Input(batch_shape=(None, 10, 8,  8, 1)))
2model.add(tf.keras.layers.Reshape((8, 8, 1), input_shape=(-1, 10, 8,  8, 1)))

試したこと

エラー内容と、その原因はわかっているのですが、解決策がわからなく困っています。

補足情報（FW/ツールのバージョンなど）

tensorflow-gpu 2.2.0
Keras 2.4.2
Python 3.6.9

行動規範の内容に同意します

あなたの回答

tips

プレビュー

行動規範の内容に同意します

質問の解決につながる回答をしましょう。サンプルコードなど、より具体的な説明があると質問者の理解の助けになります。また、読む側のことを考えた、分かりやすい文章を心がけましょう。

まだ回答がついていません

会員登録して回答してみよう

アカウントをお持ちの方は

15分調べてもわからないことは
teratailで質問しよう！

ただいまの回答率
85.48%

質問をまとめることで
思考を整理して素早く解決

テンプレート機能で
簡単に質問をまとめる

質問する

質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう！