adding optimization of multiplication by adjoint

Author: racinmat

src/onehot.jl

@@ -231,6 +231,11 @@ function Base.:(*)(A::AbstractMatrix, B::OneHotMatrix{<:Any, L}) where L
   return NNlib.gather(A, B.indices)
 end
 
+function Base.:(*)(A::AbstractMatrix, B::Adjoint{Bool, <:OneHotMatrix{<:Any, L}}) where L
+  size(A, 2) == L || throw(DimensionMismatch("Matrix column must correspond with OneHot size: $(size(A, 2)) != $L"))
+  return NNlib.scatter(+, A, parent(B).indices, dstsize=(size(A,1), size(B,2)))
+end
+
 for wrapper in [:Adjoint, :Transpose]
   @eval begin
     function Base.:*(A::$wrapper{<:Any, <:AbstractMatrix{T}}, b::OneHotVector{<:Any, L}) where {L, T}

test/onehot.jl

@@ -54,6 +54,7 @@ end
   @test A * b1' == A * Array(b1')
   @test transpose(X) * b2 == transpose(X) * Array(b2)
   @test_throws DimensionMismatch A*b2
+  @test_throws DimensionMismatch A*b2'
 end
 
 @testset "OneHotArray" begin