You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Ideally someone extends the rule. But until then, can you reshape before & after to fit the single dims case? Or, less efficiently, make a comprehension which just calls view.
With Julia 1.9
eachslice
enables to properly drop dimension when taking a reduction function over some dimension (at least based on JuliaLang/julia#16606). However, having implemented this, it seems multiple dimensions is not handled yet. I stumbled upon @mcabbott advice, "the gradient rule foreachslice
is unable to handle case right now, please make an issue" to fill an issue, so here it is. If I'm correct, this has not been opened yet.The text was updated successfully, but these errors were encountered: