fix broadcasting

xukai92 · Apr 22, 2019 · bfb7e73 · bfb7e73
1 parent 2a4d3f2
commit bfb7e73
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/distributions/gumbel.jl b/src/distributions/gumbel.jl
@@ -93,7 +93,7 @@ function logpdf(bgb::BatchGumbelBernoulli, x; τ=FT(0.2))
     α = bgb.p ./ (1 .- bgb.p .+ _eps)
     xstabe = x .+ _eps
     omxstabe = 1 .- x .+ _eps
-    return log(τ) .+ log(α) + (-τ - 1) * (log.(xstabe) + log.(omxstabe)) - 2 * (log(α .* xstabe.^(-τ) + omxstabe.^(-τ) .+ _eps))
+    return log(τ) .+ log.(α) + (-τ - 1) * (log.(xstabe) + log.(omxstabe)) - 2 * (log.(α .* xstabe.^(-τ) + omxstabe.^(-τ) .+ _eps))
 end
 
 """