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 > | |
Flatten & | apply () |
Flatten & | apply (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 |
Definition at line 90 of file VariableTypeUtils.h.