几行代码实现torch推理加速