model output has different size for dims and reshape for reshape_elementcount3