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

A Collation for Example<Tensor, NoTarget> types that stacks all data tensors into one tensor. More...

#include <stack.h>

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

Public Member Functions
TensorExample	apply_batch (std::vector< TensorExample > 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< TensorExample >

A Collation for Example<Tensor, NoTarget> types that stacks all data tensors into one tensor.

Definition at line 36 of file stack.h.

---

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

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