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 Functional API

• https://pytorch.dev.org.tw/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