IndexOrder#

产品支持情况#

  • Ascend 950PR/Ascend 950DT:支持

  • Atlas A3 训练系列产品/Atlas A3 推理系列产品:不支持

  • Atlas A2 训练系列产品/Atlas A2 推理系列产品:不支持

功能说明#

IndexOrder定义了vf.arange生成的索引序列方向,用于控制序列是递增还是递减。

原型定义#

class IndexOrder(enum.Enum):
     INCREASE_ORDER = ...  # 递增序列:dst[i] = start + i(默认)
     DECREASE_ORDER = ...  # 递减序列:dst[i] = start - i

调用示例#

import pypto_pro.language as pl

@pl.vector_function
def vf_kernel():
    dst = vf.arange(start=0, order=pl.IndexOrder.INCREASE_ORDER)