We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
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
axis
where
Feature request: Add the following arguments:
-1
None
to the following loss functions:
convex_kl_divergence
cosine_distance
cosine_similarity
kl_divergence
sigmoid_focal_loss
softmax_cross_entropy
softmax_cross_entropy_with_integer_labels
I can submit a PR for this.
The text was updated successfully, but these errors were encountered:
Hello @carlosgmartin,
Yes, that would be great. Thanks for catching this!
Sorry, something went wrong.
Great. Once #898 is merged I'll put together a PR.
No branches or pull requests
Feature request: Add the following arguments:
axis
: reduction axis (default-1
)where
: reduction mask (defaultNone
)to the following loss functions:
convex_kl_divergence
cosine_distance
cosine_similarity
kl_divergence
sigmoid_focal_loss
softmax_cross_entropy
softmax_cross_entropy_with_integer_labels
I can submit a PR for this.
The text was updated successfully, but these errors were encountered: