昇腾社区首页
中文
注册

infer(tensor, args)

函数功能

通过输入多个Tensor对象进行模型推理。

输入参数说明

参数名

类型

说明

tensor

Tensor对象

模型推理所需Tensor。

如需输入多个Tensor,请使用”,”以分隔形式传入,最大传入Tensor数量为1024。

args

可变参数

返回参数说明

返回模型推理输出的List[base.Tensor]。

抛异常接口

参数校验失败,分配Tensor内存失败及推理失败,抛出runtime异常。