Enhance the unit test framework to explicitly test whether the operator correctly handles gradients for multiple inputs.