Update src/onehot.jl

Co-authored-by: Carlo Lucibello <carlo.lucibello@gmail.com>

Author: racinmat

src/onehot.jl

@@ -227,7 +227,6 @@ function Base.:(*)(A::AbstractMatrix, B::OneHotLike{<:Any, L}) where L
 end
 
 function Base.:(*)(A::AbstractMatrix, B::OneHotMatrix{<:Any, L}) where L
-  _isonehot(B) || return invoke(*, Tuple{AbstractMatrix, AbstractMatrix}, A, B)
   size(A, 2) == L || throw(DimensionMismatch("Matrix column must correspond with OneHot size: $(size(A, 2)) != $L"))
   return NNlib.gather(A, B.indices)
 end