jax.jacobian

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

是 jax.jacrev() 的别名。

参数:

• fun (可调用对象)

• argnums (int | 序列[int])

• has_aux (bool)

• holomorphic (bool)

• allow_int (bool)

返回类型:

可调用对象