Dense layer probably produces InvalidArgumentError: Incompatible shapes: [0,2] vs. [32,2]

Question

The setup

I am using Python 3.6, TF 2.4.0 on an Azure DSVM STANDARD_NC6 (6 Cores, 56 GB RAM, 380 GB Disk) using 1 GPU

---

The paramete