Rate this Page

Template Class BatchTransform#

Inheritance Relationships#

Derived Types#

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.