From 9425e5060286891a012a7809ecd344682d30f403 Mon Sep 17 00:00:00 2001
From: Michael Abbott <32575566+mcabbott@users.noreply.github.com>
Date: Sat, 7 Jan 2023 12:02:52 -0500
Subject: [PATCH] fix a typo

---
 src/dropout.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/dropout.jl b/src/dropout.jl
index a04222672..d021ee54d 100644
--- a/src/dropout.jl
+++ b/src/dropout.jl
@@ -63,7 +63,7 @@ dropout!(B::AbstractArray, A::AbstractArray, p::Real; dims = :) = dropout!(_rng_
 function dropout!(rng::AbstractRNG, dst::AbstractArray, src::AbstractArray, p::Real; dims=:)
     size(dst) == size(src) || throw(DimensionMismatch("dropout! expects output array the same size as input"))
     0 <= p <= 1 || throw(ArgumentError("dropout expects a probability 0 <= p <= 1"))
-    _rng_compat_array(rng, A)
+    _rng_compat_array(rng, src)
     if p > 0
         pT = convert(real(eltype(dst)), p)
         _dropout!(rng, dst, src, pT, dims)