Skip to content

Commit 6ee06bc

Browse files
committed
Revert "[distribution.py] Add keepdims option for gaussian log_probs"
This reverts commit a2b4c8a.
1 parent 5964a98 commit 6ee06bc

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

chainerrl/distribution.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -276,13 +276,13 @@ def sample(self):
276276
def prob(self, x):
277277
return F.exp(self.log_prob(x))
278278

279-
def log_prob(self, x, keepdims=False):
279+
def log_prob(self, x):
280280
# log N(x|mean,var)
281281
# = -0.5log(2pi) - 0.5log(var) - (x - mean)**2 / (2*var)
282282
log_probs = -0.5 * np.log(2 * np.pi) - \
283283
0.5 * self.ln_var - \
284284
((x - self.mean) ** 2) / (2 * self.var)
285-
return F.sum(log_probs, axis=1, keepdims=keepdims)
285+
return F.sum(log_probs, axis=1)
286286

287287
@cached_property
288288
def entropy(self):

0 commit comments

Comments
 (0)