|
lamppp
|
Public Member Functions | |
| ClampBackward (tensor::Scalar min_val, tensor::Scalar max_val) | |
| variable_list | apply (const variable_list &gradOutputs) override |
Public Member Functions inherited from lmp::autograd::Function | |
| variable_list | operator() (const variable_list &inputs) |
Public Attributes | |
| tensor::Scalar | min_val_ |
| tensor::Scalar | max_val_ |
Public Attributes inherited from lmp::autograd::Function | |
| std::unique_ptr< variable_list > | saved_inputs |
|
overridevirtual |
Implements lmp::autograd::Function.