Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
T< Target > Struct Template Reference
Inheritance diagram for T< Target >:
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 > > >

Public Member Functions

torch::Tensor operator() (torch::Tensor input) override
 Transforms a single input tensor to an output tensor.
 
- Public Member Functions inherited from torch::data::transforms::TensorTransform< Target >
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.
 

Additional Inherited Members

- Public Types inherited from torch::data::transforms::TensorTransform< Target >
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 > >
 

Detailed Description

template<typename Target>
struct T< Target >

Definition at line 482 of file dataloader.cpp.


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