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