Fix torch.jit.script() export for pruned_transducer_stateless2 (#1410)

Fix torch.jit.script() export for pruned_transducer_stateless2 (#1410) #738