Struct ExtractVariables

• Defined in File custom_function.h

Inheritance Relationships

Base Type

• public IterArgs< ExtractVariables >

Struct Documentation

struct ExtractVariables : public IterArgs<ExtractVariables>

Public Functions

inline ExtractVariables(std::vector<bool> &is_var, variable_list &list)

inline void operator()(const c10::optional<at::Tensor> &x)

inline void operator()(const at::Tensor &x)

inline void operator()(const at::TensorList &list)

template<typename T>
inline void operator()(const T &x)

Public Members

std::vector<bool> &is_var_

variable_list &list_