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.

Sign up for GitHub

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

Jump to bottom

Add axis and where arguments to loss functions. #912

Open

carlosgmartin wants to merge 1 commit into google-deepmind:main from carlosgmartin:loss_axis_where

Contributor

carlosgmartin commented Apr 8, 2024 •

edited


          Add axis and where arguments to loss functions.

797dae8

carlosgmartin force-pushed the loss_axis_where branch from 12fafb3 to 797dae8 Compare

April 8, 2024 03:02

Collaborator

vroulet commented Apr 8, 2024

Thanks @carlosgmartin, could you add tests?
Also you will need to wait for #916 to pass.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment