Struct ExtractVariables¶
Defined in File custom_function.h
Page Contents
Struct Documentation¶
-
struct ExtractVariables : public IterArgs<ExtractVariables>¶
Public Functions
-
inline ExtractVariables(std::vector<bool> &is_var, variable_list &list)¶
-
inline void operator()(const at::TensorList &list)¶
-
inline ExtractVariables(std::vector<bool> &is_var, variable_list &list)¶