jax.ops 模块

在 JAX 0.2.22 中被弃用的函数 jax.ops.index_update、jax.ops.index_add 等已被移除。请改用 JAX 数组上的 jax.numpy.ndarray.at 属性。

段落缩减操作符

segment_max(data, segment_ids[, ...])	计算数组中各段的最大值。
segment_min(data, segment_ids[, ...])	计算数组中各段的极小值。
segment_prod(data, segment_ids[, ...])	计算数组中各段内的乘积。
segment_sum(data, segment_ids[, ...])	计算数组中各段的和。