Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Types | Public Member Functions
torch::data::transforms::TensorTransform< Target > Class Template Referenceabstract

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

#include <tensor.h>

Inheritance diagram for torch::data::transforms::TensorTransform< Target >:
torch::data::transforms::Transform< Example< Tensor, Target >, Example< Tensor, Target > > torch::data::transforms::BatchTransform< std::vector< Example< Tensor, Target > >, std::vector< Example< Tensor, Target > > > T< Target > torch::data::transforms::Normalize< Target > torch::data::transforms::TensorLambda< Target >

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.
 

Detailed Description

template<typename Target = Tensor>
class torch::data::transforms::TensorTransform< Target >

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.

Definition at line 18 of file tensor.h.


The documentation for this class was generated from the following file: