Calculate the Jacobian batch-wise of multiple input neural network with complex weights

It is able to differentiate such output and one can get the Jacobian as described in https://discuss.pytorch.org/t/computing-batch-jacobian-efficiently/80771/6.