Hi @sdkyuan,
I think you just got the wrong dimensions for the labels, they should be (2, 6) and not (2, 4).
Check this other post out: Shapes (2, 4) and (2, 6) are incompatible
Also, please don’t show your code in the post. You can always send it as a DM to a mentor or just show the errors, as to not spoil the answer to others
Thanks and hope that helps!