CANN/pypto的arange函数API文档

pypto.arange【免费下载链接】pyptoPyPTO发音: pai p-t-oParallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto产品支持情况产品是否支持Ascend 950PR/Ascend 950DT√Atlas A3 训练系列产品/Atlas A3 推理系列产品√Atlas A2 训练系列产品/Atlas A2 推理系列产品√功能说明创建长度为$\left\lceil \frac{\text{end} - \text{start}}{\text{step}} \right\rceil$的一维Tensor包含区间 [start, end) 内、以 step 为步长的等差数列。函数原型arange(start: Union[int, float] 0, end: Union[int, float], step: Union[int, float] 1) - Tensor参数说明参数名输入/输出说明start输入源操作数。支持的数据类型为DT_FP16DT_BF16DT_INT16DT_INT32DT_FP32。默认值为 0。end输入源操作数。支持的数据类型为DT_FP16DT_BF16DT_INT16DT_INT32DT_FP32。该参数不能省略。step输入源操作数。支持的数据类型为DT_FP16DT_BF16DT_INT16DT_INT32DT_FP32。默认值为 1。返回值说明返回一维输出Tensor若输入值存在float数据类型则输出Tensor数据类型为float否则为int。约束说明step不能为0作为浮点数abs(step)1e-8(end-start)/step需大于0如果 start, end, step 均为 int 输入则三者均不能超出 int32 范围调用示例TileShape设置示例调用该operation接口前应通过set_vec_tile_shapes设置TileShape。TileShape和输出output维度一致均为一维。如输入start为mend为n, step为p, 输出shape为[q], TileShape设置为[q1], 则q1分别用于切分q轴。pypto.set_vec_tile_shapes(16)接口调用示例y1 pypto.arange(1.0, 4.0, 0.5) y2 pypto.arange(1.0, 4.0) y3 pypto.arange(4)结果示例如下输出数据y1: [1.0, 1.5, 2.0, 2.5, 3.0, 3.5] 输出数据y2: [1.0, 2.0, 3.0] 输出数据y3: [0, 1, 2, 3]【免费下载链接】pyptoPyPTO发音: pai p-t-oParallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考