在网络中插入自己定义的layer时，如何获取前一层的输出作为自定义层的输入？

你好，非常感谢您的分享。

我遇到一个问题，我在实现了自定义层之后，准备将其嵌入原有网络，但是在tensorrt中我该如何获取前一层的输出作为自定义layer的输入呢？