resolve domain

xukai92 · Apr 23, 2019 · 74a7da2 · 74a7da2
1 parent 6b0f3ef
commit 74a7da2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/distributions/gumbel.jl b/src/distributions/gumbel.jl
@@ -65,7 +65,7 @@ BatchGumbelBernoulli(p; τ=FT(0.2)) = BatchGumbelBernoulli(p, τ)
 Sample from Gumbel-Bernoulli distributions.
 """
 function rand(gb::BatchGumbelBernoulli{T}) where {T}
-    _T = isa(gb.p, AutoGrad.Result) ? typeof(value(gb)) : T
+    _T = isa(gb.p, AutoGrad.Result) ? typeof(Knet.value(gb)) : T
 
     # TODO: re-implement this `rand` using the same procedure for `BatchGumbelBernoulliLogit`
     FT = eltype(gb.p)