added unit interval domain map

2026-06-27 18:06:19 +08:00 · 2021-05-18 08:26:11 +02:00
parent 4c0a13c448
commit 0e085ac3a9
1 changed files with 1 additions and 0 deletions
@@ -515,6 +515,7 @@ class ImplicitQuantileOutput(IndependentDistributionOutput):
        available_domain_map_cls = {
            "Positive": nn.Softplus,
            "Real": nn.Identity,
+            "Unit": nn.Softmax,
        }
        assert (
            output_domain in available_domain_map_cls.keys()