变量查询接口,获取Session内所有variable算子或指定variable算子的tensor内容。
数据类型为string的接口后续版本会废弃,建议使用数据类型为非string的接口。
Status GetVariables(const std::vector<std::string> &var_names, std::vector<Tensor> &var_values)
Status GetVariables(const std::vector<AscendString> &var_names, std::vector<Tensor> &var_values)
参数名 |
输入/输出 |
描述 |
---|---|---|
var_names |
输入 |
variable算子的名字,为空时返回所有variable类型算子的值。 |
var_values |
输出 |
variable算子的值,按照tensor格式返回;与var_names保持一致序列。 |
参数名 |
类型 |
描述 |
---|---|---|
- |
Status |
SUCCESS: 成功 其他:不存在该name的variable算子;variable未执行初始化等场景 |
无。
Atlas 推理系列产品(Ascend 310P处理器),支持
Atlas 训练系列产品,支持
Atlas A2训练系列产品,支持
Atlas 200/300/500 推理产品,不支持
Atlas 200/500 A2推理产品,不支持