https://cerwinvegassalee.blogspot.com/2023/01/logits-and-labels-mismatch-in-tensorflow.html