ak.autograd.elementwise_grad

ak.autograd.elementwise_grad(fun)

Decorator for differentiating a function that is applied elementwise to a NumPy or Awkward Array.

See Autograd for an example of autograd.elementwise_grad. This version of the decorator allows ak.Array as arguments to the function, not just NumPy arrays.

To do: examples.