接口说明

KTFOP提供的C++接口如表1所示。

表1 接口说明

算子

功能

Select算子

根据条件张量的真假,从两个输入张量中选择元素,构成一个新的张量。

Less算子

比较两个输入张量,逐元素返回一个布尔张量,表示第一个输入是否小于第二个输入

Greater算子

比较两个输入张量,逐元素返回一个布尔张量,表示第一个输入是否大于第二个输入。

FloorMod算子

计算两个输入张量的逐元素模运算

Matmul算子

执行两个矩阵的乘法运算。

LookupTableFind算子

在查找表中查找输入张量的元素,并返回对应的值。

SetExternalLogFunc接口

支持传入日志接口,打印参数校验失败信息、函数注册信息。