model output 'output' has different size for dims and reshape for reshape_variable3