Generalize input/output checks from #5990 #5993
Description
It would be more general to check that either inputs_ size equals ctx.getNumInputs() or that the last inputs_ is variadic.
Originally posted by @gramalingam in #5990 (comment)
{{ message }}