jax.numpy.insert

jax.numpy.insert(arr, obj, values, axis=None)

在给定索引之前沿给定轴插入值。

numpy.insert() 的 LAX 后端实现。

原始文档字符串如下。

参数:

• arr (array_like) – 输入数组。

• obj (int, slice or sequence of ints) – 定义插入 values 之前索引或索引的对象。

• values (array_like) – 要插入 arr 中的值。如果 values 的类型与 arr 的类型不同，则 values 将转换为 arr 的类型。 values 的形状应该使得 arr[...,obj,...] = values 是合法的。

• axis (int, 可选) – 要插入 values 的轴。如果 axis 为 None，则先将 arr 展平。

返回值:

out – 带有插入 values 的 arr 的副本。请注意，insert 不会就地进行：它会返回一个新的数组。如果 axis 为 None，则 out 是一个扁平化的数组。

返回类型:

ndarray