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

TemporaryVariable

产品支持情况

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

功能说明

创建一个临时变量张量。TemporaryVariable 使用完毕后,需将变量张量的引用传递给配对的 DestroyTemporaryVariable 算子进行销毁。根据shape和dtype属性推导输出张量y的形状和数据类型:输出形状与shape属性一致,输出数据类型与dtype属性一致。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
shape 属性 必选。变量张量的形状。 ListInt -
dtype 属性 必选。变量张量中元素的数据类型。 Int -
var_name 属性 可选。要创建的变量名称。 String,默认值为"" -
y 输出 创建的变量张量,形状与shape属性一致,数据类型与dtype属性一致。 TensorType::ALL() ND

约束说明

  • 本算子为实验性功能(EXPERIMENTAL),请勿在生产环境使用。
  • 本算子为 TensorFlow 兼容算子,图引擎将其注册为图优化阶段应被消除的节点(GeDeletedOp):真实 TensorFlow 图中由框架解析器配对 DestroyTemporaryVariable 在解析阶段处理,因此本算子无法通过 RunGraph/BuildModel 独立编译执行。

调用说明

TemporaryVariable算子为图引擎基础算子,与DestroyTemporaryVariable配对使用:本算子创建临时变量并供下游写入,DestroyTemporaryVariable在所有使用完成后读取终值并销毁临时变量。

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