mindspore.ops.slice_scatter

查看源文件
mindspore.ops.slice_scatter(input, src, axis=0, start=None, end=None, step=1)[源代码]

沿指定轴将源tensor嵌入到切片后的目标tensor。

参数:
  • input (Tensor) - 输入tensor。

  • src (Tensor) - 用于嵌入到 input 的源tensor。

  • axis (int,可选) - 指定轴,默认 0

  • start (int,可选) - 指定轴中,嵌入的开始索引,默认 None ,表示 start0

  • end (int,可选) - 指定轴中,嵌入的结束索引,默认 None ,表示 endinput 在指定维度的长度。

  • step (int,可选) - 嵌入时跳过的步长,默认 1

返回:

Tensor

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> a = mindspore.ops.zeros((4, 6))
>>> b = mindspore.ops.ones((4, 3))
>>> output = mindspore.ops.slice_scatter(input=a, src=b, axis=1, start=0, end=5, step=2)
>>> print(output)
[[1. 0. 1. 0. 1. 0.]
 [1. 0. 1. 0. 1. 0.]
 [1. 0. 1. 0. 1. 0.]
 [1. 0. 1. 0. 1. 0.]]