04lm40

The size of tensor a (16) must match the size of tensor b (4) at non-singleton dimension 0

2

Hello, I met a problem while running，“The size of tensor a (16) must match the size of tensor b (4) at non-singleton dimension 0.” How can I solve it? thank...