Expand description
Differentiable structured-prediction layers.
Currently the sinkhorn_crf module: entropy-regularised optimal-transport
(Sinkhorn) normalisation as a differentiable replacement for sum-product
normalisation in matching / permutation structured prediction.
Re-exports§
pub use sinkhorn_crf::SinkhornConfig;pub use sinkhorn_crf::SinkhornCrf;pub use sinkhorn_crf::SinkhornResult;pub use sinkhorn_crf::sinkhorn_normalize;pub use sinkhorn_crf::sinkhorn_normalize_with_margins;
Modules§
- sinkhorn_
crf - Sinkhorn CRF: entropy-regularised optimal-transport normalisation for structured prediction.