Shortcuts

Struct ExtractVariables

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 std::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_

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources