Add rrule for efficient copy constructor

Jutho · Sep 29, 2023 · 3139ae7 · 3139ae7
1 parent 1fe4955
commit 3139ae7
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/ext/TensorKitChainRulesCoreExt.jl b/ext/TensorKitChainRulesCoreExt.jl
@@ -53,6 +53,11 @@ function ChainRulesCore.rrule(::typeof(convert), T::Type{<:Array}, t::AbstractTe
     return A, convert_pullback
 end
 
+function ChainRulesCore.rrule(::typeof(Base.copy), t::AbstractTensorMap)
+    copy_pullback(Δt) = NoTangent(), Δt
+    return copy(t), copy_pullback
+end
+
 # Base Linear Algebra
 # -------------------