RefData
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR&950DT系列产品 | √ |
| Atlas A3系列产品 | √ |
| Atlas A2系列产品 | √ |
| Atlas 200I/500 A2推理产品 | √ |
| Atlas推理系列产品 | √ |
| Atlas训练系列产品 | √ |
功能说明
为其他算子提供可被引用覆盖的输入数据。RefData算子作为图中的数据输入节点,将输入张量x直接传递给输出张量y,同时通过index属性标识该数据节点在网络中的序号。与Data算子不同,RefData可被ref类算子覆盖,起到变量的作用。
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| x | 输入 | 输入张量。 | TensorType::ALL() | ND |
| index | 属性 | 输入张量的索引。数据类型为int32或int64。假设网络中有两个data节点和一个ref_data节点,前两个data的index设为0和1,ref_data的index应设为2。 | Int,默认值为0 | - |
| y | 输出 | 与输入张量x具有相同形状和数据类型的张量 | TensorType::ALL() | ND |
约束说明
无
调用说明
RefData算子为图引擎基础算子,通常作为图中数据输入节点使用,不单独调用。
| 调用方式 | 调用样例 | 说明 |
|---|---|---|
| 图模式调用 | test_geir_ref_data | 通过算子IR构图方式调用RefData算子 |