Skip to content
Snippets Groups Projects

Add BCE loss function

Merged Olivier Antoni requested to merge oantoni/aidge_learning:BCE_loss into dev
1 unresolved thread
Files
5
@@ -31,6 +31,8 @@ namespace loss {
*/
Tensor MSE(std::shared_ptr<Tensor>& prediction,
const std::shared_ptr<Tensor>& target);
Tensor BCE(std::shared_ptr<Tensor>& prediction,
const std::shared_ptr<Tensor>& target);
} // namespace loss
} // namespace Aidge
Loading