Autograd multiple inputs