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