From 3a8db785f81f4dedc6598d2fbb69ac8a2917e7b5 Mon Sep 17 00:00:00 2001 From: Alfred Duncan Date: Thu, 12 Aug 2021 11:54:30 +0100 Subject: [PATCH] Updated ChainRulesCore.rrule to replace the pullback for axes argument with NoTangent --- src/compat/chainrulescore.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/compat/chainrulescore.jl b/src/compat/chainrulescore.jl index 052ce57e..be876232 100644 --- a/src/compat/chainrulescore.jl +++ b/src/compat/chainrulescore.jl @@ -10,4 +10,4 @@ end ChainRulesCore.rrule(::typeof(getdata), x::ComponentArray) = getdata(x), Δ->(ChainRulesCore.NoTangent(), ComponentArray(Δ, getaxes(x))) -ChainRulesCore.rrule(::Type{ComponentArray}, data, axes) = ComponentArray(data, axes), Δ->(ChainRulesCore.NoTangent(), getdata(Δ), getaxes(Δ)) +ChainRulesCore.rrule(::Type{ComponentArray}, data, axes) = ComponentArray(data, axes), Δ->(ChainRulesCore.NoTangent(), getdata(Δ), ChainRulesCore.NoTangent())