jax.jacobian#

jax.jacobian(fun, argnums=0, has_aux=False, holomorphic=False, allow_int=False)[source]#

别名：jax.jacrev()。

参数:

• fun (Callable)

• argnums (int | Sequence[int])

• has_aux (bool)

• holomorphic (bool)

• allow_int (bool)

返回类型:

Callable