Skip to content

Commit 9425e50

Browse files
authored
fix a typo
1 parent 4832bb8 commit 9425e50

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/dropout.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ dropout!(B::AbstractArray, A::AbstractArray, p::Real; dims = :) = dropout!(_rng_
6363
function dropout!(rng::AbstractRNG, dst::AbstractArray, src::AbstractArray, p::Real; dims=:)
6464
size(dst) == size(src) || throw(DimensionMismatch("dropout! expects output array the same size as input"))
6565
0 <= p <= 1 || throw(ArgumentError("dropout expects a probability 0 <= p <= 1"))
66-
_rng_compat_array(rng, A)
66+
_rng_compat_array(rng, src)
6767
if p > 0
6868
pT = convert(real(eltype(dst)), p)
6969
_dropout!(rng, dst, src, pT, dims)

0 commit comments

Comments
 (0)