more overloads for StaticArrays

JuliaDiff · Jan 19, 2024 · 7eb435b · 7eb435b
1 parent b642a9f
commit 7eb435b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/extra_rules.jl b/src/extra_rules.jl
@@ -178,6 +178,7 @@ function ChainRules.frule((_, ∂x), ::Type{SArray{S, T, N, L}}, x::NTuple{L,T})
 end
 
 Base.view(t::Tangent{T}, inds) where T<:SVector = view(T(ChainRulesCore.backing(t.data)), inds)
+Base.getindex(t::Tangent{<:SVector, <:NamedTuple}, ind::Int) = ChainRulesCore.backing(t.data)[ind]
 
 function ChainRules.frule((_, ∂x), ::Type{SArray{S, T, N, L}}, x::NTuple{L,Any}) where {S, T, N, L}
     SArray{S, T, N, L}(x), SArray{S}(∂x)