复数张量切片

复数操作

复数张量切片

复数张量切片(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