RefExit
产品支持情况
| 产品 |
是否支持 |
| Ascend 950PR&950DT系列产品 |
√ |
| Atlas A3系列产品 |
√ |
| Atlas A2系列产品 |
√ |
| Atlas 200I/500 A2推理产品 |
√ |
| Atlas推理系列产品 |
√ |
| Atlas训练系列产品 |
√ |
功能说明
退出当前帧(frame)到其父帧。RefExit算子与Enter算子配合使用在图中创建循环,将可被引用覆盖的输入张量x传递到父帧使其可用。与Exit算子不同,RefExit作用于可被ref类算子覆盖的引用张量。输出张量y与输入张量x具有相同的形状和数据类型。
参数说明
| 参数名 |
输入/输出/属性 |
描述 |
数据类型 |
数据格式 |
| x |
输入 |
输入张量。需要传递到父帧的可被引用覆盖的张量。 |
float16、float32、float64、int8、int16、int32、int64、uint8、uint16、uint32、uint64、bool |
ND |
| y |
输出 |
与输入张量x具有相同形状和数据类型的张量。 |
float16、float32、float64、int8、int16、int32、int64、uint8、uint16、uint32、uint64、bool |
ND |
约束说明
调用说明
RefExit算子为控制流算子,通常与Enter算子配合使用在图中创建循环,不单独调用。