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