SliceWrite

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	√
Atlas 推理系列产品	√
Atlas 训练系列产品	√

功能说明

• 算子功能：将 value tensor 写入 x tensor 的指定位置（由 begin 指定偏移）。这是一个原地操作（in-place operation），输出与输入 x 共用同一块内存。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	目标 tensor，将被原地修改。	FLOAT16、FLOAT、DOUBLE、INT32、INT64	ND
begin	输入	写入起始位置偏移量。1D tensor，最多 2 个元素（row_offset, col_offset）。	INT32、INT64	ND
value	输入	要写入的 tensor。维度必须与 x 相同。	FLOAT16、FLOAT、DOUBLE、INT32、INT64	ND
x	输出	输出结果，与输入 x 同一块内存（原地操作）。	FLOAT16、FLOAT、DOUBLE、INT32、INT64	ND

约束说明

• x 维度必须 <= 2
• value 维度必须与 x 维度相同
• begin 必须是 1D tensor，元素个数 <= 2
• value shape + begin offset 必须在 x shape 范围内
• 输入 x 和输出 x 必须是同一块内存

调用说明

调用方式	样例代码	说明
图模式接口	test_geir_slice_write	通过算子IR接口方式调用SliceWrite算子。