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)