A Transform
that is specialized for the typical Example<Tensor, Tensor>
combination.
More...
#include <tensor.h>
Public Types | |
using | E = Example< Tensor, Target > |
Public Types inherited from torch::data::transforms::Transform< Example< Tensor, Target >, Example< Tensor, Target > > | |
using | InputType = Example< Tensor, Target > |
using | OutputType = Example< Tensor, Target > |
Public Types inherited from torch::data::transforms::BatchTransform< std::vector< Example< Tensor, Target > >, std::vector< Example< Tensor, Target > > > | |
using | InputBatchType = std::vector< Example< Tensor, Target > > |
using | OutputBatchType = std::vector< Example< Tensor, Target > > |
Public Member Functions | |
virtual Tensor | operator() (Tensor input)=0 |
Transforms a single input tensor to an output tensor. | |
OutputType | apply (InputType input) override |
Implementation of Transform::apply that calls operator() . | |
Public Member Functions inherited from torch::data::transforms::Transform< Example< Tensor, Target >, Example< Tensor, Target > > | |
std::vector< Example< Tensor, Target > > | apply_batch (std::vector< Example< Tensor, Target > > input_batch) override |
Applies the transformation over the entire input_batch . | |
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.