昇腾社区首页
中文
注册
开发者
下载

EsCreateEsCTensor

产品支持情况

产品

是否支持

Atlas A3 训练系列产品/Atlas A3 推理系列产品

Atlas A2 训练系列产品/Atlas A2 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

头文件/库文件

  • 头文件:#include <ge/esb_funcs.h>
  • 库文件:libeager_style_graph_builder_base.so、libeager_style_graph_builder_base_static.a

功能说明

本接口用于C用户创建EsCTensor。

函数原型

1
EsCTensor *EsCreateEsCTensor(const void *data, const int64_t *dim, int64_t dim_num, C_DataType data_type,C_Format format)

参数说明

参数名

输入/输出

说明

data

输入

张量数据指针。

dim

输入

张量维度数组指针。

dim_num

输入

张量维度数量。

data_type

输入

张量的DataType枚举值。

format

输入

张量格式。

返回值说明

参数名

类型

说明

-

EsCTensor *

张量的匿名指针,失败时返回nullptr。

约束说明

调用示例

1
2
3
4
std::vector<float> data = {5.0, 6.1};
int64_t dims[] = {2};
auto tmp = data.data();
auto es_tensor = EsCreateEsCTensor(tmp, dims, 1, C_DT_FLOAT, C_FORMAT_ND);