Rules for `repeat` - Githubissues

JuliaDiff / ChainRules.jl

forward and reverse mode automatic differentiation primitives for Julia Base + StdLibs

Other

435 stars 89 forks source link

Rules for `repeat` #400

Closed mzgubic closed 3 years ago

mzgubic commented 3 years ago

We only have repeat rules defined for https://github.com/JuliaDiff/ChainRules.jl/blob/38caf4bfdb8af616fcbe7626d10699608af21904/src/rulesets/Base/nondiff.jl#L56

Zygote has adjoints for repeat, an issue came up recently about them https://github.com/FluxML/Zygote.jl/issues/950 so took the opportunity to open this one.

mzgubic commented 3 years ago

Duplicate of https://github.com/JuliaDiff/ChainRules.jl/issues/383