Model(modelPath, deviceId=0)
bytes_to_ptr(data)
函数功能
将二进制数据转换成指针地址值。
输入参数说明
| 参数名 | 类型 | 说明 |
|---|---|---|
| data | py::buffer二进制数据类型 | 传二进制数据。 |
返回参数说明
返回指针地址值。
父主题： 媒体数据处理
transpose(input_tensor, axes)
函数功能
可通过指定的一组轴维度（axes），对输入的Tensor数据进行转置处理，如未指定具体轴维度，则默认对Tensor数据进行反序转置。
功能仅支持在Host侧执行操作，如需处理Device侧Tensor数据，请先通过to_host接口，将Device侧数据转移到Host侧，再进行转置。
输入参数说明
| 参数名 | 类型 | 说明 |
|---|---|---|
| input_tensor | Tensor类 | 待转置Tensor类。 维度支持2维、3维、4维，数据类型支持base.dtype.float32、base.dtype.float16、base.dtype.uint8，具体请参见Tensor类的属性列表。 |
| axes | List[int] | 转置选项，默认值为空。 如果未指定具体axes，则默认生成反序“axes”对“input”中的数据进行反序转置。 例如：三维张量默认生成反序“axes”为{2, 1, 0} 。 |
返回参数说明
返回转置后的Tensor类数据。
父主题： 媒体数据处理
Model(modelPath, deviceId=0)
函数功能
初始化模型。
输入参数说明
| 参数名 | 类型 | 说明 |
|---|---|---|
| modelPath | string | 模型路径。 |
| deviceId | int | 模型部署的芯片ID，默认为0号芯片。（-1表示模型部署在Host侧，为保留字段，请勿使用。） |
返回参数说明
模型对象
父主题： Model类
Model(option, deviceId)
函数功能
用户可选择特定加载方式，加载推理模型。
输入参数说明
| 参数名 | 类型 | 说明 |
|---|---|---|
| option | ModelLoadOptV2数据类型 | 用户可通过参数选择模型输入方式，具体请参见ModelLoadOptV2类。 |
| deviceId | int | 模型部署的芯片ID，默认为0号芯片。（-1表示模型部署在Host侧，为保留字段，请勿使用。） |
返回参数说明
返回模型对象。
父主题： Model类
infer(tensorList)
函数功能
通过输入Tensor列表进行模型推理。
输入参数说明
| 参数名 | 类型 | 说明 |
|---|---|---|
| tensorList | List[base.Tensor] | Tensor列表。 |
返回参数说明
返回模型推理输出的List[base.Tensor]。
抛异常接口
分配Tensor内存失败及推理失败，抛出runtime异常。
父主题： Model类
infer(tensor, args)
函数功能
通过输入多个Tensor对象进行模型推理。
输入参数说明
参数名 | 类型 | 说明 |
|---|---|---|
tensor | Tensor对象 | 模型推理所需Tensor。 如需输入多个Tensor，请使用”,”以分隔形式传入，最大传入Tensor数量为1024。 |
args | 可变参数 |
返回参数说明
返回模型推理输出的List[base.Tensor]。
抛异常接口
参数校验失败，分配Tensor内存失败及推理失败，抛出runtime异常。
父主题： Model类
在线提单