https://mingredbestcheapprice.blogspot.com/2024/06/keras-output-shape-has-extra-dimension.html