lamppp
|
Public Member Functions | |
ReshapeBackward (std::vector< size_t > shape) | |
variable_list | apply (const variable_list &gradOutputs) override |
![]() | |
variable_list | operator() (const variable_list &inputs) |
Public Attributes | |
std::vector< size_t > | shape |
![]() | |
std::unique_ptr< variable_list > | saved_inputs |
|
overridevirtual |
Implements lmp::autograd::Function.