Solved by ptrblck in post #4 
              
              
                I’m not sure where this error is coming from as your model definition isn’t posted. 
Check the complete stack trace and see, if you can spot the line of code which raises the error. 
Based on the posted error message e.g. a matrix multiplication might fail with the shape mismatch e.g. in a linear la…