Fix logic error in checks

FluxML · Jun 15, 2021 · a52d343 · a52d343
1 parent 7576e8f
commit a52d343
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/onehot.jl b/src/onehot.jl
@@ -190,14 +190,14 @@ end
 for wrapper in [:Adjoint, :Transpose]
   @eval begin
     function Base.:*(A::$wrapper{<:Any, <:AbstractMatrix{T}}, b::OneHotVector{<:Any, L}) where {L, T}
-      size(A, 2) != L ||
+      size(A, 2) == L ||
         throw(DimensionMismatch("Matrix column must correspond with OneHot size: $(size(A, 2)) != $L"))
 
       return A[:, b.ix]
     end
 
     function Base.:*(A::$wrapper{<:Number, <:AbstractVector{T}}, b::OneHotVector{<:Any, L}) where {L, T}
-      size(A, 2) != L ||
+      size(A, 2) == L ||
         throw(DimensionMismatch("Matrix column must correspond with OneHot size: $(size(A, 2)) != $L"))
 
       return A[b.ix]