Fix TorchDeterministic (#9241)

2026-07-09 10:37:48 +08:00 · 2020-07-23 19:43:20 +02:00
parent 9f969260e8
commit 5303c3abe3
1 changed files with 1 additions and 1 deletions
@@ -308,7 +308,7 @@ class TorchDeterministic(TorchDistributionWrapper):

    @override(TorchDistributionWrapper)
    def sampled_action_logp(self):
-        return 0.0
+        return torch.zeros((self.inputs.size()[0], ), dtype=torch.float32)

    @override(TorchDistributionWrapper)
    def sample(self):