Merge pull request optuna#4712 from not522/speedup-logpdf

Speed up `logpdf` in `_truncnorm.py`
Alnusjaponica · May 29, 2023 · a031e95 · a031e95
2 parents 58b57eb + 27fbd30
commit a031e95
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/optuna/samplers/_tpe/_truncnorm.py b/optuna/samplers/_tpe/_truncnorm.py
@@ -225,10 +225,10 @@ def logpdf(
     x = (x - loc) / scale
 
     x, a, b = np.atleast_1d(x, a, b)
-    x, a, b = np.broadcast_arrays(x, a, b)
 
     out = _norm_logpdf(x) - _log_gauss_mass(a, b)
 
+    x, a, b = np.broadcast_arrays(x, a, b)
     out[(x < a) | (b < x)] = -np.inf
     out[a == b] = math.nan