mlx.core.put_along_axis

put_along_axis(a: array, /, indices: array, values: array, axis: int | None = None, *, stream: None | Stream | Device = None) → array

在指定索引处沿轴放置值。

参数：

• a (array) – 目标数组。

• indices (array) – 索引数组。除了 axis 维度，该数组应可与输入数组广播。

• values (array) – 值数组。该数组应可与索引广播。

• axis (int 或 None) – 在目标数组中放置值的轴。如果 axis == None，则在放置操作之前将目标数组展平。

返回值：

输出数组。

返回类型：

array