put_along_axis¶

参数¶

• arr (Tensor) - 输入的 Tensor 作为目标矩阵，数据类型为：float32、float64。

• indices (Tensor) - 索引矩阵，包含沿轴提取 1d 切片的下标，必须和 arr 矩阵有相同的维度，需要能够 broadcast 与 arr 矩阵对齐，数据类型为：int、int64。

• value （float）- 需要插入的值，形状和维度需要能够被 broadcast 与 indices 矩阵匹配，数据类型为：float32、float64。

• axis (int) - 指定沿着哪个维度获取对应的值，数据类型为：int。

• reduce (str，可选) - 归约操作类型，默认为 assign，可选为 add 或 multiple。不同的规约操作插入值 value 对于输入矩阵 arr 会有不同的行为，如为 assgin 则覆盖输入矩阵，add 则累加至输入矩阵，multiple 则累乘至输入矩阵。

返回¶

输出 Tensor，indeces 矩阵选定的下标会被插入 value，与 arr 数据类型相同。

代码示例¶

import paddle

x = paddle.to_tensor([[10, 30, 20], [60, 40, 50]])
index = paddle.to_tensor([[0]])
value = 99
axis = 0
result = paddle.put_along_axis(x, index, value, axis)
print(result)
# [[99, 99, 99],
# [60, 40, 50]]