torchprof - 用于Pytorch模型的逐层分析的最小依赖库