For
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR&950DT系列产品 | √ |
| Atlas A3系列产品 | √ |
| Atlas A2系列产品 | √ |
| Atlas 200I/500 A2推理产品 | √ |
| Atlas推理系列产品 | √ |
| Atlas训练系列产品 | √ |
功能说明
- 算子功能:循环执行
body子图,循环变量从start开始,每次迭代递增delta,直到循环变量超过limit。每次迭代时,将input传入body子图,body返回的输出作为下一次迭代的输入,最终输出为最后一次迭代的结果。
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
start |
输入 | 循环下界,int32标量。 | INT32 | ND |
limit |
输入 | 循环上界,int32标量。 | INT32 | ND |
delta |
输入 | 步长大小,int32标量。 | INT32 | ND |
input |
输入(动态) | 输入张量,支持动态输入。 | 所有类型 | ND |
body |
子图 | 接收input并返回另一组张量的子图,用于计算下一次迭代的值。 |
- | - |
output |
输出(动态) | body子图返回的输出张量,类型与input相同,支持动态输出。 |
所有类型 | ND |
约束说明
start、limit、delta必须为INT32标量。input的数量必须与output的数量一致。- 各
input与对应output的数据类型需保持一致。
调用说明
| 调用方式 | 调用样例 | 说明 |
|---|---|---|
| 图模式调用 | test_geir_for | 通过算子IR构图方式调用For算子。 |