https://github.com/kmkurn/pytorch-crf/blob/1b7ced20c01352ccf70b7ae7bb99a58366de8e48/torchcrf/init.py#L259 26
It’s pretty deal with this function, i move out this function as a non-class function and add @torch.jit.script