Template Class Transform

• Defined in File base.h

Inheritance Relationships

Base Type

• public torch::data::transforms::BatchTransform< std::vector< Input >, std::vector< Output > > (Template Class BatchTransform)

Class Documentation

template<typename Input, typename Output>
class Transform : public torch::data::transforms::BatchTransform<std::vector<Input>, std::vector<Output>>

A transformation of individual input examples to individual output examples.

Just like a Dataset is a BatchDataset, a Transform is a BatchTransform that can operate on the level of individual examples rather than entire batches. The batch-level transform is implemented (by default) in terms of the example-level transform, though this can be customized.

Public Types

using InputType = Input

using OutputType = Output

Public Functions

virtual OutputType apply(InputType input) = 0

Applies the transformation to the given input.

inline virtual std::vector<Output> apply_batch(std::vector<Input> input_batch) override

Applies the transformation over the entire input_batch.