更新时间:2021-03-18 GMT+08:00
功能描述
本样例使用TIK方式进行了matmul算子的实现,给定两个输入的张量A和B,进行矩阵相乘,输出结果张量。
说明:此样例未实现算子适配插件,若开发者想在原始框架为caffe的网络模型中使用此自定义算子,需要自定义实现对应的算子适配插件。
父主题: Matmul算子(TIK方式)
