Shortcuts

Struct MarginRankingLossOptions

Page Contents

Struct Documentation

struct torch::nn::MarginRankingLossOptions

Options for the MarginRankingLoss module.

Example:

MarginRankingLoss model(MarginRankingLossOptions().margin(0.5).reduction(torch::kSum));

Public Types

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

Public Functions

auto margin(const double &new_margin) -> decltype(*this)

Has a default value of 0.

auto margin(double &&new_margin) -> decltype(*this)
const double &margin() const noexcept
double &margin() noexcept
auto reduction(const reduction_t &new_reduction) -> decltype(*this)

Specifies the reduction to apply to the output. Default: Mean.

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

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources