search

FluxML / ZygoteRules.jl

MIT License
15 stars 13 forks source link

Calling Zygote.forward from within rules #2

Closed willtebbutt closed 5 years ago

willtebbutt commented 5 years ago

You can't currently do this in ZygoteRules as forward isn't defined at this point. I imagine the fix would be straightforward enough?

MikeInnes commented 5 years ago

Yes, we could just declare that function in this package.