torch::autograd::Scatter Struct Reference

Inheritance diagram for torch::autograd::Scatter:

Public Member Functions
	Scatter (std::vector< at::Device > devices, const c10::optional< std::vector< int64_t >> &chunk_sizes=c10::nullopt, int64_t dim=0, const c10::optional< std::vector< c10::optional< at::cuda::CUDAStream >>> &streams=c10::nullopt, bool unsqueeze_scalars=false)
variable_list	apply (variable_list &&inputs) override

Data Fields
std::vector< at::Device >	devices_
c10::optional< std::vector< int64_t > >	chunk_sizes_
int64_t	dim_
c10::optional< std::vector< c10::optional< at::cuda::CUDAStream > > >	streams_
bool	unsqueeze_scalars_

Detailed Description

Definition at line 16 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