torch::data::transforms::TensorTransform< Target > Class Template Reference

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

#include <tensor.h>

Inheritance diagram for torch::data::transforms::TensorTransform< 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:

• torch/csrc/api/include/torch/data/transforms/tensor.h