GetDynamicGearInfo
GetOutputTensorDataType
函数功能
获得模型输出的对应Tensor的数据类型信息。
函数原型
MxBase::TensorDType GetOutputTensorDataType(uint32_t index = 0)
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| index | 输入 | 模型输出的索引(第index个输出)。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| MxBase::TensorDType | Tensor的数据类型,具体请参见TensorDType。 |
父主题: Model
GetInputFormat
函数功能
获得模型输入的数据组织形式(NHWC 或者 NCHW)。
函数原型
MxBase::VisionDataFormat GetInputFormat()
返回参数说明
| 数据结构 | 说明 |
|---|---|
| MxBase::VisionDataFormat | 模型的输入数据组织形式,具体请参见VisionDataFormat。 |
父主题: Model
GetDynamicGearInfo
函数功能
获得模型支持的动态输入档位信息,支持动态Batch、动态分辨率和分档动态维度模型。
函数原型
std::vector<std::vector<uint64_t>> GetDynamicGearInfo()
返回参数说明
| 数据结构 | 说明 |
|---|---|
| std::vector<std::vector<uint64_t>> | 模型支持的输入Tensor档位信息。 - **std::vector<std::vector<uint64_t>>**为所有档位信息。 - **std::vector<uint64_t>**为某一档的具体值。 返回数据结构为空,表示获取分档信息失败或者模型不属于函数功能中描述的三种模型之一,具体的原因可查看报错信息确认。 例如,模型为动态分辨率模型,**std::vector<uint64_t>**为某一档的高、宽值。 |
父主题: Model
公共参数
| 参数名 | 数据类型 | 参数说明 |
|---|---|---|
| stream | void* | Stream对应的指针,默认值为“nullptr”。 |
| isDefault | bool | 声明当前创建的Stream是否为默认Stream,默认值为“false”(即非默认)。 |
父主题: AscendStream
AscendStream
函数功能
AscendStream类的构造函数。
函数原型
explicit AscendStream(int32_t deviceId = 0);
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| deviceId | 输入 | 指定Stream对应的设备ID。 |
父主题: AscendStream
~AscendStream
函数功能
AscendStream类的默认析构函数。
函数原型
~AscendStream();
父主题: AscendStream
在线提单