此外,我们的目标是所有非公共 API 应以下划线作为前缀命名,尽管我们目前还未完全遵守这一规定。
未覆盖的内容是什么?
任何以下划线开头的内容。
jax....向量化 (vmap)
vmap(fun[, in_axes, out_axes, axis_name, …]) 向量化映射。...numpy.vectorize(pyfunc, *[, excluded, signature]) 定义一个支持广播的向量化函数。...inexact() 所有数值标量类型的抽象基类,其值的表示(可能)是不精确的,如浮点数。
inner(a, b, *[, precision, …]) 计算两个数组的内积。...与 numpy.ndarray 一样,大多数用户不需要手动实例化 Array 对象,而是通过 jax.numpy 函数如 array()、arange()、linspace() 和上面列出的其他函数来创建它们