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:

• torch/csrc/autograd/functions/comm.h
• torch/csrc/autograd/functions/comm.cpp