fix policy gradients for mujoco domains (#589)

2026-07-12 11:59:20 +08:00 · 2017-05-24 18:39:37 -07:00
parent c5bc76193f
commit 679910496e
1 changed files with 1 additions and 1 deletions
@@ -42,7 +42,7 @@ class Categorical(object):
 class DiagGaussian(object):
  def __init__(self, flat):
    self.flat = flat
-    mean, logstd = tf.split(1, 2, flat)
+    mean, logstd = tf.split(flat, 2, axis=1)
    self.mean = mean
    self.logstd = logstd
    self.std = tf.exp(logstd)