model output 'output' has different number of variable-size dimensions for dims and reshape for reshape_nobatch_variable5