model output has different size for dims and reshape for reshape_variable1