Add axis parameter to dice loss

[td-jakubl]

#19637

[google-cla]

Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).

View this failed invocation of the CLA check for more information.

For the most up to date status, view the checks section at the bottom of the pull request.

[gbaned]

Hi @td-jakubl Can you please sign CLA. Thank you!

[td-jakubl]

Hi @td-jakubl Can you please sign CLA. Thank you!

Done :)

[td-jakubl]

Hi @gbaned some test fails and perhaps this is not related to changes in this PR. Do you have any idea why?

[fchollet]

Thanks for the PR! Please add a unit test.

[fchollet]

Please use a code block for the code example. Do not use TF APIs, only Keras APIs (backend agnostic)

[td-jakubl]

Updated

[fchollet]

LGTM, thank you for the contribution!

[codecov-commenter]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 66.69%. Comparing base (669a97a) to head (2501601).
Report is 1 commits behind head on master.

Additional details and impacted files

@@             Coverage Diff             @@
##           master   #19673       +/-   ##
===========================================
- Coverage   81.47%   66.69%   -14.79%     
===========================================
  Files         498      498               
  Lines       45773    45757       -16     
  Branches     8458     8454        -4     
===========================================
- Hits        37294    30517     -6777     
- Misses       6577    13595     +7018     
+ Partials     1902     1645      -257     

Flag	Coverage Δ
keras	66.61% <100.00%> (-14.71%)	⬇️
keras-jax	?
keras-numpy	56.29% <100.00%> (-0.16%)	⬇️
keras-tensorflow	?
keras-torch	61.99% <100.00%> (-2.95%)	⬇️
keras.applications	?
keras.applications-jax	?
keras.applications-numpy	?
keras.applications-tensorflow	?
keras.applications-torch	?

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[td-jakubl]

LGTM, thank you for the contribution!

No, problem. I reformated code. Could you rerun workflows?