tensorRT教程——tensor RT OP理解(实现自定义层,搭建网络)
首先如果你的自定义操作可以通过一些矩阵操作来实现,那么你大可不必自己去通过plug in的方式实现,可以使用tensor RT 的OP来组合实现。 他的OP极其类似tensor flow的操作,如果看到不理解,那就去谷歌tensor flow的操作,触类旁通即可,但仍有一些文档描述错误。 一些我使用过典型的OP备注如下,并附上一个测试OP的脚本,按需修改即可:1、reduce层的理解: reduce主要是在各个轴上进行对应的操作。 axes...