select_scatter

参数

• x (Tensor) - 输入的 Tensor 作为目标矩阵，数据类型为： bool、 float16、 float32、 float64、 uint8、 int8、 int16、 int32、 int64、 bfloat16、 complex64、 complex128。别名 input。

• values (Tensor) - 需要插入的值，形状需要与 x 矩阵除去第 axis 维后的形状一致，数据类型为： bool、 float16、 float32、 float64、 uint8、 int8、 int16、 int32、 int64、 bfloat16、 complex64、 complex128。别名 src。

• axis (int) - 指定沿着哪个维度嵌入对应的值，数据类型为：int。别名 dim。

• index (int) - 指定沿着 axis 维的哪一列嵌入对应的值，数据类型为：int。

返回

输出 Tensor， x 矩阵的第 axis 维的第 index 列会被插入 value，与 x 数据类型相同。

代码示例

>>> import paddle

>>> x = paddle.zeros((2, 3, 4)).astype("float32")
>>> values = paddle.ones((2, 4)).astype("float32")
>>> res = paddle.select_scatter(x, values, 1, 1)
>>> print(res)
Tensor(shape=[2, 3, 4], dtype=float32, place=Place(cpu), stop_gradient=True,
[[[0., 0., 0., 0.],
  [1., 1., 1., 1.],
  [0., 0., 0., 0.]],
 [[0., 0., 0., 0.],
  [1., 1., 1., 1.],
  [0., 0., 0., 0.]]])