fix bmm gradient

Author: chengchingwen

src/batchedmul.jl

@@ -39,28 +39,27 @@ function ∇batched_mul(Δ::AbstractArray{T, 3}, A::AbstractArray{T, 3}, B::Abst
     (batched_mul(Δ, batched_transpose(B)), batched_mul(batched_transpose(A), Δ))
 end
 
-
 function ∇batched_mul(Δ::AbstractArray{T, 3}, A::BatchedTranspose{T, <: AbstractArray{T, 3}}, B::AbstractArray{T, 3}) where T
-    (batched_mul(Δ, batched_transpose(B)), batched_mul(A, Δ))
+    (batched_mul(B, batched_transpose(Δ)), batched_mul(batched_transpose(A), Δ))
 end
 
 function ∇batched_mul(Δ::AbstractArray{T, 3}, A::AbstractArray{T, 3}, B::BatchedTranspose{T, <: AbstractArray{T, 3}}) where T
-    (batched_mul(Δ, B), batched_mul(batched_transpose(A), Δ))
+    (batched_mul(Δ, batched_transpose(B)), batched_mul(batched_transpose(Δ), A))
 end
 
 function ∇batched_mul(Δ::AbstractArray{T, 3}, A::BatchedTranspose{T, <: AbstractArray{T, 3}}, B::BatchedTranspose{T, <: AbstractArray{T, 3}}) where T
-    (batched_mul(batched_transpose(Δ), batched_transpose(B)), batched_mul(batched_transpose(A), batched_transpose(Δ)))
+    (batched_mul(batched_transpose(B), batched_transpose(Δ)), batched_mul(batched_transpose(Δ), batched_transpose(A)))
 end
 
 function ∇batched_mul(Δ::AbstractArray{T, 3}, A::BatchedAdjoint{T, <: AbstractArray{T, 3}}, B::AbstractArray{T, 3}) where T
-    (batched_mul(Δ, batched_adjoint(B)), batched_mul(A, Δ))
+    (batched_mul(B, batched_adjoint(Δ)), batched_mul(batched_adjoint(A), Δ))
 end
 
 function ∇batched_mul(Δ::AbstractArray{T, 3}, A::AbstractArray{T, 3}, B::BatchedAdjoint{T, <: AbstractArray{T, 3}}) where T
-    (batched_mul(Δ, B), batched_mul(batched_adjoint(A), Δ))
+    (batched_mul(Δ, batched_adjoint(B)), batched_mul(batched_adjoint(Δ), A))
 end
 
 function ∇batched_mul(Δ::AbstractArray{T, 3}, A::BatchedAdjoint{T, <: AbstractArray{T, 3}}, B::BatchedAdjoint{T, <: AbstractArray{T, 3}}) where T
-    (batched_mul(batched_adjoint(Δ), batched_adjoint(B)), batched_mul(batched_adjoint(A), batched_adjoint(Δ)))
+    (batched_mul(batched_adjoint(B), batched_adjoint(Δ)), batched_mul(batched_adjoint(Δ), batched_adjoint(A)))
 end