tenzr::nn::BCELoss¶

Callable wrapper around the binary cross-entropy loss.

#include <loss.h>

Public Functions¶

	Name
	BCELoss(double eps =1.0e-7, std::string reduction ="mean")
Tensor	operator()(const Tensor & pred, const Tensor & target) const
Tensor	Forward(const Tensor & pred, const Tensor & target) const

inline explicit BCELoss(
    double eps =1.0e-7,
    std::string reduction ="mean"
)

inline Tensor operator()(
    const Tensor & pred,
    const Tensor & target
) const

inline Tensor Forward(
    const Tensor & pred,
    const Tensor & target
) const

Return: Same as operator().