JAX,它们如何将库集成到其 API 中,它在数学上添加了什么功能,并且如何在其他库中用于计算加速。...., 238.59848022,
245.62597656, 240.22348022]], dtype=float32)
对于在 CPU 上进行的 1000x1000 矩阵乘法来说,269µs 的时间是一个令人惊讶地小的时间...isdtype(dtype, kind) 返回一个布尔值,指示提供的 dtype 是否属于指定的 kind。
isfinite(x, /) 测试每个元素是否有限(既不是无穷大也不是非数)。...nan_to_num(x[, copy, nan, posinf, neginf]) 将 NaN 替换为零,将无穷大替换为大的有限数(默认
nanargmax(a[, axis, out, keepdims...(a[, axis, out, keepdims, initial, where]) 返回数组或指定轴上的最大值,忽略任何 NaN
nanmean(a[, axis, dtype, out, keepdims