Fix Tensor constructor rrule on Thunks

bsc-quantic · Sep 12, 2023 · 89dfb77 · 89dfb77
1 parent 78ec48c
commit 89dfb77
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ext/TenetChainRulesCoreExt.jl b/ext/TenetChainRulesCoreExt.jl
@@ -13,6 +13,7 @@ ChainRulesCore.frule((_, Δ, _), T::Type{<:Tensor}, data, inds; meta...) = T(dat
 
 function ChainRulesCore.rrule(T::Type{<:Tensor}, data, inds; meta...)
     Tensor_pullback(Δ) = (NoTangent(), Δ.data, NoTangent())
+    Tensor_pullback(Δ::Thunk) = (NoTangent(), unthunk(Δ).data, NoTangent())
     return T(data, inds; meta...), Tensor_pullback
 end