lamppp
Loading...
Searching...
No Matches
Public Member Functions | Public Attributes | List of all members
lmp::autograd::ops::ReshapeBackward Struct Reference
Inheritance diagram for lmp::autograd::ops::ReshapeBackward:
lmp::autograd::Function

Public Member Functions

 ReshapeBackward (std::vector< size_t > shape)
 
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

std::vector< size_t > shape
 
- Public Attributes inherited from lmp::autograd::Function
std::unique_ptr< variable_list > saved_inputs
 

Member Function Documentation

◆ apply()

variable_list lmp::autograd::ops::ReshapeBackward::apply ( const variable_list &  gradOutputs)
overridevirtual

The documentation for this struct was generated from the following file: