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