-
Notifications
You must be signed in to change notification settings - Fork 3
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
slicemap on TrackedArrays produces Array{TrackedReal} #3
Comments
Oops, I see now that this function is really only supported for Zygote. Feel free to close this issue if appropriate. |
That’s right, I don’t recally exactly why but I couldn’t make the gradients work in general for Tracker. (For The simplest version of this was the |
However you can do this:
I guess anything with |
This is a problem since the code below will not run on the GPU (unless one allows scalar operations which is not ideal)
The text was updated successfully, but these errors were encountered: