Problem with keras.backend.batch_dot

[alitk]

I used to clone and run this repo, and it worked fine. I wrote a code based on your implementation of the capsule layer. However, today, when I run the code, I get this weird error. I tracked it, and it looks like there is a problem with batch_dot. Do you have any idea how to fix it? Even this repository after clone and run gives me the same error — both on Google Colab and my local machine.
Thanks

ValueError: Can not do batch_dot on inputs with shapes (None, 10, 10, 1152, 16) and (None, 10, None, 1152, 16) with axes=[2, 3]. x.shape[2] != y.shape[3] (10 != 1152).