Template Class TensorTransform

• Defined in File tensor.h

Inheritance Relationships

Base Type

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

Class Documentation

template<typename Target = Tensor>
class TensorTransform : public torch::data::transforms::Transform<Example<Tensor, Tensor>, Example<Tensor, Tensor>>

A Transform that is specialized for the typical Example<Tensor, Tensor> combination.

It exposes a single operator() interface hook (for subclasses), and calls this function on input Example objects.

Public Types

using E = Example<Tensor, Target>

Public Functions

virtual Tensor operator()(Tensor input) = 0

Transforms a single input tensor to an output tensor.

inline virtual OutputType apply(InputType input) override

Implementation of Transform::apply that calls operator().