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

Public Member Functions

 Gather (const at::Device &destination_device, int64_t dim=0)
variable_list apply (variable_list &&inputs) override

Data Fields

at::Device destination_device_
int64_t dim_

Detailed Description

Definition at line 34 of file comm.h.

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