Skip to content

Commit 31c71ee

Browse files
committed
view of Eye returns a OneElement
1 parent 67fc055 commit 31c71ee

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

src/oneelement.jl

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,4 +48,11 @@ Base.replace_in_print_matrix(o::OneElement{<:Any,2}, k::Integer, j::Integer, s::
4848
function Base.setindex(A::Zeros{T,N}, v, kj::Vararg{Int,N}) where {T,N}
4949
@boundscheck checkbounds(A, kj...)
5050
OneElement(convert(T, v), kj, axes(A))
51-
end
51+
end
52+
53+
54+
Base.@propagate_inbounds function view(A::RectOrDiagonal{<:Any,<:AbstractFill}, kr::AbstractRange, j::Integer)
55+
@boundscheck checkbounds(A, kr, j)
56+
k = findfirst(isequal(j), kr)
57+
OneElement(getindex_value(A.diag), isnothing(k) ? 0 : something(k), length(kr))
58+
end

0 commit comments

Comments
 (0)