Inheritance diagram for lmp::autograd::ops::Reshape:
Public Types | |
| using | DefaultBackward = ReshapeBackward |
Public Member Functions | |
| Reshape (std::vector< size_t > shape) | |
| tensor::Tensor | execute (const variable_list &inputs) const |
 Public Member Functions inherited from lmp::autograd::ForwardFunction< Reshape > | |
| variable_list | apply (const variable_list &inputs) override |
| variable_list | apply (const variable_list &inputs, Args &&... args) |
| 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 |
Additional Inherited Members | |
| Static Public Member Functions inherited from lmp::autograd::ForwardFunction< Reshape > | |
| static bool | requires_grad (const variable_list &variables) |
The documentation for this struct was generated from the following file:
- include/lamppp/autograd/functions/view_ops.hpp
