Jutho/TensorOperations.jl

Does TensorOperations.jl not support Zygote.jl?

fuyangfeng opened this issue · 1 comments

julia> using Zygote

julia> using TensorOperations

julia> function f(x) 
        @tensor b[i,k] := x[i,j]*x[j,k]
        sum(b)
        end
f (generic function with 1 method)

julia> a=rand(10,10);

julia> gradient(f,a)
ERROR: Can't differentiate gc_preserve_end expression $(Expr(:gc_preserve_end, %77)).
You might want to check the Zygote limitations documentation.
https://fluxml.ai/Zygote.jl/latest/limitations

Just reran this example with 1.3.1. Same error. Is Zygote support a WIP? Thanks