fix(rllib): allow explore=False with tuple action distributions (#10443)

2026-07-13 17:45:08 +08:00 · 2020-09-11 00:03:02 +02:00
parent 7aa55ca82c
commit 504da45e69
1 changed files with 1 additions and 1 deletions
@@ -72,5 +72,5 @@ class StochasticSampling(Exploration):
            logp = action_dist.sampled_action_logp()
        else:
            action = action_dist.deterministic_sample()
-            logp = torch.zeros((action.size()[0], ), dtype=torch.float32)
+            logp = torch.zeros_like(action_dist.sampled_action_logp())
        return action, logp