变换

eval(*args)	评估一个array或array的树。
compile(fun[, inputs, outputs, shapeless])	返回一个编译后的函数，该函数产生与fun相同的输出。
custom_function	设置一个用于自定义梯度和vmap定义的函数。
disable_compile()	全局禁用编译。
enable_compile()	全局启用编译。
grad(fun[, argnums, argnames])	返回一个计算fun梯度的函数。
value_and_grad(fun[, argnums, argnames])	返回一个计算fun值和梯度的函数。
jvp(fun, primals, tangents)	计算雅可比向量积。
vjp(fun, primals, cotangents)	计算向量-雅可比积。
vmap(fun[, in_axes, out_axes])	返回fun的向量化版本。