复数张量切片
复数操作
复数张量切片
复数张量切片(Complex Tensor Slicing)算子沿多个轴生成输入张量的切片。使用轴、起始标记和结尾标记属性来指定轴列表中每个轴的起点和终点位置,并使用此信息来对输入张量切片。
参数列表
| 参数 | 类型 | 参数描述 |
|---|---|---|
| 输入张量 | 输入 | 多维张量 |
| 轴 | 输入 | 表示进行切片的轴 |
| 起始标记 | 输入 | 表示在各个轴上切片的起始索引值 |
| 结尾标记 | 输入 | 表示在各个轴上切片的结束索引值 |
| 输出张量 | 输出 | 多维张量 |
函数返回值
- 没有错误:操作成功。
- 对象未初始化:表示输入张量没有初始化。
- 类型不匹配:表示参数的数据类型不一致。
- 分配失败:表张量分配空间不足。
- 内部错误:表示内部调用出错。
注意:如果向起始标记或结尾标记传递负值如−i,则表示该轴的反向第 i−1个位置(这里以 0 为初始位置)。如果传递给起始标记或结尾标记的值大于 n(维度中的元素数目),则表示 n。当切片一个未知数量的维度时,建议传入整形最大值。轴、起始标记和结尾标记三个参数的元素数目应相等。
参考来源:团体标准——《人工智能 科学计算 共性算子 功能要求》——章节7.3.2.11
硬件验证状态
昇腾 NPU 1.8ms鲲鹏 CPU 2.6ms海光 CPU 2.1ms昆仑 XPU 2.4ms海光 DCU 2.3ms飞腾 CPU 3.1ms燧原 XPU 2.7ms沐曦 GPU 2.5ms
