сохранение результатов модели в Input 0 слоя несовместимо со слоем: его ранг не определен, но для слоя требуется определенный ранг

Question

Я использую tenorflow 2.1.0

Код для воспроизведения ошибки можно найти здесь. https://colab.research.google.com/drive/1Bf6EhjmnjzsZQUoiGAi1XVxjrFOPa2Cp?authuser=1

Как мне решить эту проблему?