model output 'output' dimension must be integer >= 1, or -1 to indicate a variable-size dimension for zerodims_output1