FilesLast commitLast update
1 month ago
1 month ago
1 month ago
1 month ago
1 month ago
1 month ago
README

RefData

产品支持情况

产品 是否支持
Ascend 950PR/Ascend 950DT √
Atlas A3 训练系列产品/Atlas A3 推理系列产品 √
Atlas A2 训练系列产品/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算子