jax.numpy.insert

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

在给定的轴上，在给定的索引之前插入值。

LAX-backend 实现的 numpy.insert()。

原始文档字符串如下。

参数:

• 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, optional) – 要沿其插入 values 的轴。如果 axis 为 None，则首先将 arr 展平。

返回:

out – 插入 values 后的 arr 副本。注意 insert 不是原地操作：返回一个新的数组。如果 axis 是 None，out 是一个展平的数组。

返回类型:

ndarray