Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Data Fields
torch::autograd::Flatten Struct Reference
Inheritance diagram for torch::autograd::Flatten:
torch::IterArgs< Flatten >

Public Member Functions

 Flatten (variable_list &out)
void operator() (const at::Tensor &x)
void operator() (at::ArrayRef< at::Tensor > xs)
- Public Member Functions inherited from torch::IterArgs< Flatten >
Flattenapply ()
Flattenapply (T &&arg, Args &&...args)
void operator() (at::ArrayRef< T > args)
void operator() (const std::vector< T > &args)
bool short_circuit ()

Data Fields

variable_list & out

Detailed Description

Definition at line 90 of file VariableTypeUtils.h.

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