Gradient

Compute the numerical gradient of a sampled function.

Abstract Signature:

Gradient(f: Tensor, varargs: Tensor, axis: int | Sequence[int] | None)

PyTorchJAX (Core)NumPy

PyTorch

API: torch.gradient

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jax.numpy.gradient

Strategy: Direct Mapping

Official Docs ↗

NumPy

API: numpy.gradient

Strategy: Direct Mapping

Official Docs ↗