Assign requires shapes of both tensors to match. lhs shape= [22,384] rhs shape= [15,384]

[Zhiwen-Liu]

请问为了解决这个报错，shape应该在哪里修改？