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

Variable

产品支持情况

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

功能说明

创建一个变量张量。Variable算子通过输入x赋值,在内部记录变量张量的值,输出y为创建的变量张量。InferShape函数返回成功但不显式设置输出描述,由框架默认推导(output跟随input)。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 用于给变量张量赋值的输入张量,调用者无需传递变量张量的值。 TensorType::ALL() ND
index 属性 输入张量的索引。 Int,默认值为0 -
value 属性 用于传递和记录变量张量的值。 Tensor,默认值为Tensor() -
container 属性 变量张量的容器。 String,默认值为"" -
shared_name 属性 变量张量的共享名称。 String,默认值为"" -
y 输出 创建的变量张量。 TensorType::ALL() ND

约束说明

无

调用说明

Variable算子为图引擎基础算子,通常作为图中变量节点使用,不单独调用。

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