CreateBoolTensor

产品支持情况

全量芯片支持。

头文件/库文件

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

功能说明

创建bool数据类型和张量形状的张量。

函数原型

std::unique_ptr<Tensor> CreateBoolTensor(const std::vector<bool> &value, const std::vector<int64_t> &dims, Format format = FORMAT_ND)

参数说明

返回值说明

约束说明

无

调用示例

1
2
3
4

EsGraphBuilder builder("graph_name");
std::vector<bool> data = {true, false, true, false};
std::vector<int64_t> dims = {4};
auto tensor = builder.CreateBoolTensor(data,dims);