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)

返回类型:

可调用对象