diff --git a/src/layers/basic.jl b/src/layers/basic.jl index 12d4e2e3..83eeee21 100644 --- a/src/layers/basic.jl +++ b/src/layers/basic.jl @@ -110,7 +110,7 @@ end (a::Dense{<:Any,W})(x::AbstractArray{T}) where {T <: Union{Float32,Float64}, W <: AbstractArray{T}} = invoke(a, Tuple{AbstractArray}, x) -(a::Dense{<:Any,W})(x::AbstractArray{<:Real}) where {T <: Union{Float32,Float64}, W <: AbstractArray{T}} = +(a::Dense{<:Any,W})(x::AbstractArray{<:AbstractFloat}) where {T <: Union{Float32,Float64}, W <: AbstractArray{T}} = a(T.(x)) """