I get an error when I want to change the stirde

[zhangdan94]

When I want to change the output size, I try to change the string size, but the S = sum (self. _ops [offset + j] (h, weights [offset + j]) for j, h in enumerate (states))
shows RuntimeError: The size of tensor a (20) must match the size of tensor B (40) at non-singleton dimension 3 error Is there any suggestions?

[quark0]

Make sure you've adjusted the strides for all operations, including those in FactorizedReduce.

[zhangdan94]

Thanks for your reply. And I'll have a try as you suggested.