只有当我们实际从主机检查数组的值时,例如通过打印它或将其转换为普通的 numpy.ndarray,JAX 才会强制 Python 代码等待计算完成。...clip([x, min, max, a, a_min, a_max]) 将数组中的值限制在给定范围内。
column_stack(tup) 将一维数组按列堆叠成二维数组。...from_dlpack(x, /, *[, device, copy]) 从实现了__dlpack__的对象创建一个 NumPy 数组。...hsplit(ary, indices_or_sections) 水平(按列)将数组分割为多个子数组。
hstack(tup[, dtype]) 按序列水平(按列)堆叠数组。...与 numpy.ndarray 一样,大多数用户不需要手动实例化 Array 对象,而是通过 jax.numpy 函数如 array()、arange()、linspace() 和上面列出的其他函数来创建它们