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: