李沐《动手学深度学习》第 12 章笔记 使用 torch.jit.script 提高计算性能 from torch import nn net = nn.Sequential(nn.Linear(1024, 64), nn.ReLU(), nn.Linear(64, 32), nn.RuLU(), nn.Linear(32, 10)) x = torch.rand(size=(1, 1024)) net(X) net = torch.jit.script(net) net(X)