torch::data::transforms::Stack< Example<> > Struct Template Reference

A Collation for Example<Tensor, Tensor> types that stacks all data tensors into one tensor, and all target (label) tensors into one tensor. More...

#include <stack.h>

Inheritance diagram for torch::data::transforms::Stack< Example<> >:

Public Member Functions
Example	apply_batch (std::vector< Example<>> examples) override
Public Member Functions inherited from torch::data::transforms::BatchTransform< InputBatch, OutputBatch >
virtual OutputBatch	apply_batch (InputBatch input_batch)=0
	Applies the transformation to the given input_batch.

Additional Inherited Members
Public Types inherited from torch::data::transforms::BatchTransform< InputBatch, OutputBatch >
using	InputBatchType = InputBatch
using	OutputBatchType = OutputBatch

Detailed Description

template<>
struct torch::data::transforms::Stack< Example<> >

A Collation for Example<Tensor, Tensor> types that stacks all data tensors into one tensor, and all target (label) tensors into one tensor.

Definition at line 20 of file stack.h.

---

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

• torch/csrc/api/include/torch/data/transforms/stack.h