nonzero_keys returns CartesianIndex

lkdvos · Nov 19, 2024 · 63bc908 · 63bc908
1 parent d6a4259
commit 63bc908
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/tensors/abstractblocktensor/sparsity.jl b/src/tensors/abstractblocktensor/sparsity.jl
@@ -6,7 +6,7 @@ nonzero_values(t::AbstractBlockTensorMap) = nonzero_values(parent(t))
 nonzero_length(t::AbstractBlockTensorMap) = nonzero_length(parent(t))
 
 nonzero_values(A::AbstractArray) = values(A)
-nonzero_keys(A::AbstractArray) = keys(A)
+nonzero_keys(A::AbstractArray) = eachindex(IndexCartesian(), A)
 nonzero_pairs(A::AbstractArray) = pairs(A)
 nonzero_length(A::AbstractArray) = length(A)