create_diff_fn.py

def create_diff_fn(fn):
        def diff_fn(*argv):
            jacobian = []
            Dual_arguments = [Dual(x, 0.) for x in argv]
            for input_arg in Dual_arguments:
                input_arg.derivative = 1.
                result = fn(*Dual_arguments)
                jacobian.append(result.derivative)
                input_arg.derivative = 0.

            return jacobian

        return diff_fn