Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Data Fields
torch::autograd::VariableInfo Struct Reference

Public Member Functions

 VariableInfo (const Variable &var)
 
Variable zeros (at::OptionalDeviceGuard &device_guard) const
 

Data Fields

at::Typetype
 
at::Device device = at::kCPU
 
std::vector< int64_t > size
 
bool requires_grad
 

Detailed Description

Definition at line 21 of file python_function.h.


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