Merge branch 'master' of github.bus.zalan.do:zalandoresearch/pytorch-ts

2026-06-27 18:06:19 +08:00 · 2021-06-09 13:49:03 +02:00
parent 5039dcf6ea 87680b18fb
commit 1a0872d110
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()