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 > > |
Definition at line 482 of file dataloader.cpp.