How to branch for TorchScript versions?

Question

I have a TorchScript and would like to branch based on the torch version:

import torch

def test_div():
    a = torch.tensor([10,20])
    b = torch.tensor([5,