File size: 461 Bytes
9dd3461 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
#pragma once
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
|