1
2
3
4
5
6
7
8
9
10
11
12
13
14
| #pragma once
|
| namespace Reduction {
|
| // NB: Keep this in sync with Reduction class in torch/nn/_reduction.py
| // These constants control the reduction behavior of loss functions.
| // Ideally, this would be a scoped enum, but jit doesn't support that
| enum Reduction {
| None, // Do not reduce
| Mean, // (Possibly weighted) mean of losses
| Sum, // Sum losses
| END
| };
| }
|
|
