Shortcuts

Struct SmoothL1LossOptions

Page Contents

Struct Documentation

struct torch::nn::SmoothL1LossOptions

Options for the SmoothL1Loss module.

Example:

SmoothL1Loss model(SmoothL1LossOptions().reduction(torch::kNone).beta(0.5));

Public Types

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

Public Functions

SmoothL1LossOptions()
SmoothL1LossOptions(torch::enumtype::kNone reduction)
SmoothL1LossOptions(torch::enumtype::kMean reduction)
SmoothL1LossOptions(torch::enumtype::kSum reduction)
auto reduction(const reduction_t &new_reduction) -> decltype(*this)

Specifies the reduction to apply to the output: ‘none’ | ‘mean’ | ‘sum’.

‘none’: no reduction will be applied, ‘mean’: the sum of the output will be divided by the number of elements in the output, ‘sum’: the output will be summed. Default: ‘mean’

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

Specifies the threshold at which to change between L1 and L2 loss. Default: 1.0.

auto beta(double &&new_beta) -> decltype(*this)
const double &beta() const noexcept
double &beta() 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