https://mingredbestcheapprice.blogspot.com/2024/03/tensorflow-label-and-logits-shapes-are.html