| from keras.src.api_export import keras_export | |
| class Reduction: | |
| AUTO = "auto" | |
| NONE = "none" | |
| SUM = "sum" | |
| SUM_OVER_BATCH_SIZE = "sum_over_batch_size" | |
| def all(cls): | |
| return (cls.AUTO, cls.NONE, cls.SUM, cls.SUM_OVER_BATCH_SIZE) | |
| def validate(cls, key): | |
| if key not in cls.all(): | |
| raise ValueError( | |
| f'Invalid Reduction Key: {key}. Expected keys are "{cls.all()}"' | |
| ) | |