文件最后提交记录最后更新时间
2 个月前
2 个月前
2 个月前
2 个月前
2 个月前
1 个月前
README

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算子。