Variable
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR&950DT系列产品 | √ |
| Atlas A3系列产品 | √ |
| 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算子 |