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
,表示 start 为0
。end (int,可选) - 指定轴中,嵌入的结束索引,默认
None
,表示 end 是 input 在指定维度的长度。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.]]