FilesLast commitLast update
1 month ago
1 month ago
1 month ago
1 month ago
1 month ago
21 days ago
README

VarHandleOp

产品支持情况

产品 是否支持
Ascend 950PR&950DT系列产品 √
Atlas A3系列产品 √
Atlas A2系列产品 √
Atlas 200I/500 A2推理产品 √
Atlas推理系列产品 √
Atlas训练系列产品 √

功能说明

创建一个指向变量资源的句柄。输出张量y为标量资源句柄(DT_RESOURCE类型),shape和dtype属性描述句柄所指向变量的形状和数据类型(而非输出本身的形状),供下游资源类算子(如ReadVariableOp、AssignVariableOp)使用。container和shared_name属性用于指定变量的容器和共享名称。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
container 属性 可选。变量所在的容器。 String,默认值为"" -
shared_name 属性 可选。变量的共享名称。 String,默认值为"" -
dtype 属性 必选。句柄所指向变量的元素数据类型。 Type -
shape 属性 可选。句柄所指向变量的形状。 ListInt,默认值为UNKNOWN_SHAPE -
y 输出 标量资源句柄(DT_RESOURCE类型)。 TensorType({DT_RESOURCE}) ND

约束说明

  • 本算子为实验性功能(EXPERIMENTAL),请勿在生产环境使用。
  • 本算子为资源句柄源节点,输出为标量 DT_RESOURCE 张量,无运行时计算语义,无法通过 RunGraph/BuildModel 独立执行。
  • shape/dtype 属性描述句柄所指向变量的形状与类型(并非输出本身的形状与类型),推导时经 InferenceContext 的 HandleShapesAndTypes 机制传递给下游资源类算子(框架 peer 传播),下游经 GetInputHandleShapesAndTypes 读取。

调用说明

VarHandleOp算子为图引擎基础算子,作为资源句柄源节点配合下游资源类算子(如ReadVariableOp、AssignVariableOp)使用,不单独调用。

调用方式 调用样例 说明
图模式调用 test_geir_var_handle_op 通过算子IR构图方式调用VarHandleOp算子