KTFOP提供的C++接口如表1所示。
算子
功能
Select算子
根据条件张量的真假,从两个输入张量中选择元素,构成一个新的张量。
Less算子
比较两个输入张量,逐元素返回一个布尔张量,表示第一个输入是否小于第二个输入。
Greater算子
比较两个输入张量,逐元素返回一个布尔张量,表示第一个输入是否大于第二个输入。
FloorMod算子
计算两个输入张量的逐元素模运算。
Matmul算子
执行两个矩阵的乘法运算。
LookupTableFind算子
在查找表中查找输入张量的元素,并返回对应的值。
SetExternalLogFunc接口
支持传入日志接口,打印参数校验失败信息、函数注册信息。