1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
| #pragma once
|
| #include <torch/arg.h>
| #include <torch/csrc/WindowsTorchApiMacro.h>
| #include <torch/types.h>
|
| namespace torch {
| namespace nn {
|
| /// Options for a L1 loss module.
| struct TORCH_API L1LossOptions {
| L1LossOptions(Reduction::Reduction reduction = Reduction::Mean)
| : reduction_(reduction) {}
|
| /// Specifies the reduction to apply to the output.
| TORCH_ARG(Reduction::Reduction, reduction);
| };
|
| } // namespace nn
| } // namespace torch
|
|
