Skip to content

tenzr::nn::MSELoss

#include <loss.h>

Public Functions

Name
MSELoss(std::string reduction ="mean")
Tensor operator()(const Tensor & pred, const Tensor & target) const
Tensor Forward(const Tensor & pred, const Tensor & target) const

Public Functions Documentation

function MSELoss

inline explicit MSELoss(
    std::string reduction ="mean"
)

function operator()

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

function Forward

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

Return: Same as operator().