| namespace at { | |
| 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 | |
| }; | |
| } // namespace Reduction | |
| } // namespace at | |