系统信息
OS平台和发行版（例如Linux Ubuntu 16.04）：Google Colab
TensorFlow版本：2.2和2.3
Python版本：3.6
描述问题
我想非急切地构建一个转换器（根据此notebook）模型，并且只有在使用tf.keras.layers.Input及其对应的输出，整个模型来构建模型时，一切才能顺利运行程序卡住了。顺便说一句，在执行块10中的for循环时，它会花费大量时间。
没有关于这种现象的错误消息，完全不知道导致这种现象的原因。
我已在Google colab中复制了此问题here。
提供在遇到问题之前执行的命令/步骤的确切顺序
我初始化了Input tf.keras.layers.Input并使用变压器模型生成输出，然后使用输入和输出来构建模型，但是整个程序都卡住了。