张量切片
高阶微分操作
张量切片
张量切片（Tensor Slicing）算子正向传播按照给定的起始位置、切片大小和步长数组，从输入张量中提取出相应的子张量。反向传播将梯度值传递回相应的输入张量的位置，其他位置的梯度值保持为零。
正向传播参数列表
| 参数 | 类型 | 参数描述 |
|---|---|---|
| 输入张量 | 输入 | 表示输入张量 |
| 起始位置数组 | 输入 | 表示每个维度提取起始位置数组，长度为输入张量的维度 |
| 切片大小数组 | 输入 | 表示每个维度提取大小数组，长度为输入张量的维度 |
| 切片步长数组 | 输入 | 表示每个维度上提取步长数组，长度为输入张量的维度 |
| 输出张量 | 输出 | 表示输出张量 |
反向传播参数列表
| 参数 | 类型 | 参数描述 |
|---|---|---|
| 输入张量 | 输入 | 正向传播输出张量的梯度 |
| 输出张量 | 输出 | 一个与输入张量同形状的新张量，它的切片位置上填充有对应的梯度值，而其他位置则填充零 |
函数返回值
- 没有错误：操作成功。
- 对象未初始化：表示输入张量没有初始化。
- 类型不匹配：表示参数的数据类型不一致。
- 分配失败：表张量分配空间不足。
- 内部错误：表示内部调用出错。
注：正向传播参考IEEE 2941.1-2022标准中的张量切片接口。
参考来源：团体标准——《人工智能 科学计算 共性算子功能要求》——章节7.3.1.22
硬件验证状态
昇腾 NPU 1.8ms鲲鹏 CPU 2.6ms海光 CPU 2.1ms昆仑 XPU 2.4ms海光 DCU 2.3ms飞腾 CPU 3.1ms燧原 XPU 2.7ms沐曦 GPU 2.5ms