A class to clip input values within specified bounds. If the parameters lower and upper are not specified, no clipping is performed. If the parameters are scalar values, then all elements of the input are clipped to the same bounds. If the parameters are arrays, then each element of the input is clipped to the corresponding bounds.
A class that moves the first dimension to the last place. Usefull for meta learners that return the predictions of various units in the batch dimension while in environment the num_unit (e.g., num_SKU) dimension is usually the last one
Type
Default
Details
remove_action_per_unit_dim
bool
False
If there is only one action per unit, the action dimension can be removed by setting this to True