jax.jacobian

jax.jacobian(fun, argnums=0, has_aux=False, holomorphic=False, allow_int=False)

別名為 jax.jacrev()。

參數:

• fun (Callable)

• argnums (int | Sequence[int])

• has_aux (bool)

• holomorphic (bool)

• allow_int (bool)

回傳類型:

Callable