lmp::autograd::ops::ExpandDimsBackward Struct Reference

Inheritance diagram for lmp::autograd::ops::ExpandDimsBackward:

Public Member Functions
	ExpandDimsBackward (size_t axis)
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
size_t	axis
Public Attributes inherited from lmp::autograd::Function
std::unique_ptr< variable_list >	saved_inputs

Member Function Documentation

apply()

variable_list lmp::autograd::ops::ExpandDimsBackward::apply ( const variable_list &  gradOutputs )

overridevirtual

Implements lmp::autograd::Function.

---

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

• include/lamppp/autograd/functions/view_ops.hpp