极智AI | python and cpp 实现 TensorRT elementWise Layer

  欢迎关注我的公众号 [极智视界]，获取我的更多笔记分享

  大家好，我是极智视界，本文介绍一下 python 和 cpp 实现 TensorRT elementWise 层。

  elementWise 算子指的是需要逐位运行的 op，具有十分丰富的元素间计算，如元素加、元素点乘、元素减、取极值等。这里结合 TensorRT 的实现来说，主要包括 python 实现 和 cpp 实现。

1 elementWise Layer python TensorRT 构建

  来看接口：

elementWise_Layer = network.add_elementwise(input0, input1, trt.ElementWiseOperation