autograd_gamma_incomplete_gamma_gradient_computation.py

1import autograd.numpy as np
2from autograd import grad
3from autograd_gamma import gammainc
4
5# Define a function using the incomplete gamma function
6def f(a):
7    return gammainc(a, 1.0)
8
9# Calculate the derivative of gammainc with respect to a at a=2.0
10grad_f = grad(f)
11
12print(f"Value at a=2.0: {f(2.0)}")
13print(f"Gradient at a=2.0: {grad_f(2.0)}")