transpose/adjoint 2 time return the same

Author: chengchingwen

src/batchedadjtrans.jl

@@ -15,7 +15,7 @@ end
 Lazy batched transpose.
 """
 batched_transpose(A::AbstractArray{T}) where T = BatchedTranspose(A)
-
+batched_transpose(A::BatchedTranspose) = A.parent
 
 """
     BatchedAdjoint{T, N, S} <: AbstractBatchedMatrix{T, N}
@@ -31,7 +31,7 @@ end
 Lazy batched adjoint.
 """
 batched_adjoint(A::AbstractArray{T, 3}) where T = BatchedAdjoint(A)
-
+batched_adjoint(A::BatchedAdjoint) = A.parent
 
 BatchedAdjoint(A) = BatchedAdjoint{Base.promote_op(adjoint,eltype(A)),typeof(A)}(A)
 BatchedTranspose(A) = BatchedTranspose{Base.promote_op(transpose,eltype(A)),typeof(A)}(A)

test/batchedmul.jl

@@ -41,4 +41,7 @@ end
     @test batched_mul(batched_adjoint(cA), batched_adjoint(cB)) == bmm_adjtest(cA, cB; adjA = true, adjB = true)
     @test batched_mul(batched_adjoint(cA), cC) == bmm_adjtest(cA, cC; adjA = true)
     @test batched_mul(cA, batched_adjoint(cA)) == bmm_adjtest(cA, cA; adjB = true)
+
+    @test batched_transpose(batched_transpose(A)) == A
+    @test batched_adjoint(batched_adjoint(cA)) == cA
 end