Skip to content

Commit d1a5b53

Browse files
committed
Add sparse_storage for Adjoint and Transpose
1 parent 3efdb89 commit d1a5b53

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

NDTensors/src/lib/SparseArrayInterface/src/abstractsparsearray/wrappedabstractsparsearray.jl

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,10 @@ function stored_indices(a::Adjoint)
1313
return Iterators.map(I -> CartesianIndex(reverse(Tuple(I))), stored_indices(parent(a)))
1414
end
1515
stored_length(a::Adjoint) = stored_length(parent(a))
16+
sparse_storage(a::Adjoint) = Iterators.map(adjoint, sparse_storage(parent(a)))
1617

1718
function stored_indices(a::Transpose)
1819
return Iterators.map(I -> CartesianIndex(reverse(Tuple(I))), stored_indices(parent(a)))
1920
end
2021
stored_length(a::Transpose) = stored_length(parent(a))
22+
sparse_storage(a::Transpose) = Iterators.map(transpose, sparse_storage(parent(a)))

0 commit comments

Comments
 (0)