极智AI | 讲解 TensorRT Constant 算子

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

  大家好，我是极智视界，本文讲解一下 TensorRT Constant 算子。

  Constant 算子是指常量层，这个算子一般是什么时候使用呢：一般当下一个算子是两矩阵乘 或 两矩阵点乘 或 两矩阵拼接等这类两头输入的算子，而某一个矩阵需要离线读取时，就需要用到 Constant 算子来构建这个离线读取的张量。以上介绍了 Constant 算子一个使用场景，下面介绍 TensorRT 中 Constant 算子的具体怎么来添加。

  在 TensorRT 中如何构建一个 Constant 算子呢，来看：

# 通过 add_constant 添加 constant 算子
constantLayer = network.add_constant([1], np.array