search

kmkurn / pytorch-crf

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

Add `batch_first` argument to `__init__` #11

Closed kmkurn closed 5 years ago

kmkurn commented 6 years ago

When batch_first is True, all input tensors are assumed to have batch_size in the first dimension.