overload for OneElement matrices

JuliaArrays · Aug 13, 2024 · 52314a5 · 52314a5
1 parent f659d9f
commit 52314a5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/FillArrays.jl b/src/FillArrays.jl
@@ -7,7 +7,7 @@ import Base: size, getindex, setindex!, IndexStyle, checkbounds, convert,
     any, all, axes, isone, iszero, iterate, unique, allunique, permutedims, inv,
     copy, vec, setindex!, count, ==, reshape, map, zero,
     show, view, in, mapreduce, one, reverse, promote_op, promote_rule, repeat,
-    parent, similar, issorted, add_sum, accumulate, OneTo
+    parent, similar, issorted, add_sum, accumulate, OneTo, permutedims
 
 import LinearAlgebra: rank, svdvals!, tril, triu, tril!, triu!, diag, transpose, adjoint, fill!,
     dot, norm2, norm1, normInf, normMinusInf, normp, lmul!, rmul!, diagzero, AdjointAbsVec, TransposeAbsVec,

diff --git a/src/oneelement.jl b/src/oneelement.jl
@@ -426,6 +426,10 @@ function Base.reshape(A::OneElement, shape::Tuple{Vararg{Int}})
     OneElement(A.val, Tuple(newcartind), shape)
 end
 
+#permute
+
+permutedims(o::OneElementMatrix) = OneElement(o.val, reverse(o.ind), o.axes)
+
 # show
 _maybesize(t::Tuple{Base.OneTo{Int}, Vararg{Base.OneTo{Int}}}) = size.(t,1)
 _maybesize(t) = t