The size of tensor must match the size of tensor b at non-singleton dimension

Part 1 (2019) Advanced (Part 1 v3)
2

Did you find out the answer? I have the same issue