I run your codes but I didn’t get the output ,instead I got the error
“The expanded size of the tensor (1) must match the existing size (3) at non-singleton dimension 1”
Can anybody tell me why and how to fix it? thank you!