Rate this Page
★★★★★
Template Struct CppNode#
Defined inFile custom_function.h
Inheritance Relationships#
Base Type#
publictorch::autograd::Node(Struct Node)
Struct Documentation#
- template<classT>
structCppNode:publictorch::autograd::Node# Public Functions
- virtualvariable_listapply(variable_list&&inputs)override#
Performs the
Node’s actual operation.
- virtualvoidrelease_variables()override#
Releases saved variables if the operation won’t be reused.
- voidsave_variables_to_ctx()#
- inlinevirtualvoidcompiled_args(CompiledNodeArgs&args)constoverride#
- inlinevirtualvariable_listapply_with_saved(constvariable_list&inputs,SwapSavedVariables&saved)override#
Public Members
- AutogradContextctx_#
- std::vector<bool>is_variable_input_#
- std::vector<VariableInfo>input_info_#
- std::vector<VariableInfo>output_info_#
- virtualvariable_listapply(variable_list&&inputs)override#
On this page
