@@ -0,0 +1,368 @@
+import torch%0Aimport sru%0A%0AD = 4%0Amodel = sru.SRU(D, D, num_layers=2)%0Amodel.eval()%0A%0Ats_model = torch.jit.script(model)%0Ats_model.save('sru_ts.pt')%0A%0Awith torch.no_grad():%0A    x = torch.ones(3, 2, D)%0A    h, c = model(x)%0A    h, c = h.view(-1), c.view(-1)%0A    print(''.join(%5B%22%7B:.4f%7D %22.format(x.item()) for x in h%5D))%0A    print(''.join(%5B%22%7B:.4f%7D %22.format(x.item()) for x in c%5D))%0A

add missing script