search

kmkurn / pytorch-crf

(Linear-chain) Conditional random field in PyTorch.
https://pytorch-crf.readthedocs.io
MIT License
938 stars 151 forks source link

Fix `_new` return type. #7

Closed kmkurn closed 5 years ago

kmkurn commented 6 years ago

It should be Union[torch.FloatTensor, torch.cuda.FloatTensor].