Incompatible shape error,Transformer net, error after max_subword_length changed from 20 to 1

Question

Tensorflow-gpu version 2.3, cuda toolkit version 10.1, cudnn version 7.6.5, Anaconda environment. The code used is the code from transformer tutorial on tensorflow official webs