Struct MSELossOptions

• Defined in File loss.h

Struct Documentation

struct torch::nn::MSELossOptions

Options for the MSELoss module.

Example:

MSELoss model(MSELossOptions(torch::kNone));

Public Types

typedef c10::variant<enumtype::kNone, enumtype::kMean, enumtype::kSum> reduction_t

Public Functions

MSELossOptions() = default

MSELossOptions(torch::enumtype::kNone reduction)

MSELossOptions(torch::enumtype::kMean reduction)

MSELossOptions(torch::enumtype::kSum reduction)

auto reduction(const reduction_t &new_reduction) -> decltype(*this)

Specifies the reduction to apply to the output.

'none' | 'mean' | 'sum'. Default: 'mean'

auto reduction(reduction_t &&new_reduction) -> decltype(*this)

const reduction_t &reduction() const noexcept

reduction_t &reduction() noexcept