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.
1.8.11