Torch-TensorRT 开发有用链接

TensorRT 可用层和预期维度

• https://docs.nvidia.com/deeplearning/sdk/tensorrt-support-matrix/index.html#layers-matrix

TensorRT C++ 文档

• https://docs.nvidia.com/deeplearning/tensorrt/api/c_api/index.html

TensorRT Python 文档（有时更容易阅读）

• https://docs.nvidia.com/deeplearning/sdk/tensorrt-api/python_api/index.html

PyTorch 函数式 API

• https://pytorch.ac.cn/docs/stable/nn.functional.html

PyTorch native_ops

• https://github.com/pytorch/pytorch/blob/master/aten/src/ATen/native/native_functions.yaml

PyTorch IR 文档

• https://github.com/pytorch/pytorch/blob/master/torch/csrc/jit/OVERVIEW.md