Template Class BatchTransform

• Defined in File base.h

Inheritance Relationships

Derived Types

• public torch::data::transforms::Transform< Example< Tensor, Tensor >, Example< Tensor, Tensor > > (Template Class Transform)

• public torch::data::transforms::Transform< Input, Input > (Template Class Transform)

• public torch::data::transforms::Stack< Example<> > (Template Struct Stack< Example<> >)

• public torch::data::transforms::Stack< TensorExample > (Template Struct Stack< TensorExample >)

Class Documentation

template<typename InputBatch, typename OutputBatch>
class BatchTransform

A transformation of a batch to a new batch.

Subclassed by torch::data::transforms::Transform< Example< Tensor, Tensor >, Example< Tensor, Tensor > >, torch::data::transforms::Transform< Input, Input >, torch::data::transforms::Stack< Example<> >, torch::data::transforms::Stack< TensorExample >

Public Types

using InputBatchType = InputBatch

using OutputBatchType = OutputBatch

Public Functions

virtual ~BatchTransform() = default

virtual OutputBatch apply_batch(InputBatch input_batch) = 0

Applies the transformation to the given input_batch.