CreateConst
产品支持情况
功能说明
创建Const算子。
函数原型
-
创建int64类型的Const算子
EsTensorHolder CreateConst(const std::vector<int64_t> &value, const std::vector<int64_t> dims) -
创建int32类型的Const算子
EsTensorHolder CreateConst(const std::vector<int32_t> &value, const std::vector<int64_t> dims) -
创建uint64类型的Const算子
EsTensorHolder CreateConst(const std::vector<uint64_t> &value, const std::vector<int64_t> dims) -
创建uint32类型的Const算子
EsTensorHolder CreateConst(const std::vector<uint32_t> &value, const std::vector<int64_t> dims) -
创建float类型的Const算子
EsTensorHolder CreateConst(const std::vector<float> &value, const std::vector<int64_t> dims) -
创建指定类型、格式和维度的Const算子(通用模板方法)
template <typename T> EsTensorHolder CreateConst(const std::vector<T> &value, const std::vector<int64_t> &dims, ge::DataType dt,ge::Format format = FORMAT_ND)
参数说明
返回值说明
约束说明
无
调用示例
-
创建int64类型的Const算子:
EsGraphBuilder builder("test_graph"); std::vector<int64_t> dims = {3}; std::vector<int64_t> vec64 = {1, 2, 3}; auto c1 = builder.CreateConst(vec64, dims); -
创建指定类型、格式和维度的Const算子:
EsGraphBuilder builder("test_graph"); std::vector<float> vecf = {1.1, 2.0, 3.2, 4.4}; std::vector<int64_t> dims = {4}; auto c1 = builder.CreateConst<float>(vecf, dims, ge::DT_FLOAT);