GetInputTensorShape
GetInputTensorNum
函数功能
获得模型的输入个数。
函数原型
uint32_t GetInputTensorNum()
返回参数说明
| 数据结构 | 说明 |
|---|---|
| uint32_t | 模型的输入个数。 |
父主题： Model
GetOutputTensorNum
函数功能
获得模型的输出个数。
函数原型
uint32_t GetOutputTensorNum()
返回参数说明
| 数据结构 | 说明 |
|---|---|
| uint32_t | 模型的输出个数。 |
父主题： Model
GetInputTensorShape
函数功能
获得模型输入的对应Tensor的数据shape信息。
动态模型场景下对应的动态维度查询结果为-1。如果要使用查询的结果直接传入Tensor构造函数构造Tensor，需要将int64_t数据转换为uint32_t数据。
函数原型
std::vector<int64_t> GetInputTensorShape(uint32_t index = 0)
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| index | 输入 | 模型输入的索引（第index个输入）。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| std::vector<int64_t> | 对应输入的Tensor的shape信息。 |
父主题： Model
GetOutputTensorShape
函数功能
获得模型输出的对应Tensor的数据shape信息。查询的结果可直接传入Tensor构造函数用来构造Tensor。
函数原型
std::vector<uint32_t> GetOutputTensorShape(uint32_t index = 0)
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| index | 输入 | 模型输出的索引（第index个输出）。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| std::vector<uint32_t> | 对应输出的Tensor的shape信息。 |
父主题： Model
GetInputTensorDataType
函数功能
获得模型输入的对应Tensor的数据类型信息。
函数原型
MxBase::TensorDType GetInputTensorDataType(uint32_t index = 0)
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| index | 输入 | 模型输入的索引（第index个输入）。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| MxBase::TensorDType | Tensor的数据类型，具体请参见TensorDType。 |
父主题： Model
在线提单