跳到主要内容

~TensorContext

~TensorContext

函数功能

TensorContext类的默认析构函数。

函数原型

~TensorContext();

父主题: TensorContext

GetInstance

函数功能

获取TensorContext单例对象。

函数原型

static std::shared_ptr<TensorContext> GetInstance();

返回参数说明

数据结构说明
std::shared_ptrTensorContext单例对象。

父主题: TensorContext

SetContext

函数功能

设置线程的Context。

函数原型

APP_ERROR SetContext(const uint32_t &deviceId);

参数说明

参数名输入/输出说明
deviceId输入线程当前的deviceId。

返回参数说明

数据结构说明
APP_ERROR程序执行返回的错误码,请参考APP_ERROR说明

父主题: TensorContext

类说明

该类主要有TensorShape数据结构的构造方法和相关的功能接口。

父主题: TensorShape

TensorShape

函数功能

TensorShape构造函数,创建TensorShape()对象。

函数原型

TensorShape() = default;
TensorShape (std::vector<T> shape);

参数说明

参数名输入/输出说明
shape输入Tensor的形状。 shape向量中各维度要求为正整数且单个或各项乘积需要小于52,428,8000(500MB),否则函数将抛出异常。‬

父主题: TensorShape

~TensorShape

函数功能

TensorShape类的默认析构函数。

函数原型

~TensorShape() = default;

父主题: TensorShape

GetDims

函数功能

获取Tensor形状的维度。

函数原型

uint32_t GetDims() const;

返回参数说明

数据结构说明
uint32_tTensor形状的维度值。

父主题: TensorShape

在线提单