torch::data::transforms::BatchTransform< InputBatch, OutputBatch > Class Template Reference

A transformation of a batch to a new batch. More...

#include <base.h>

Inheritance diagram for torch::data::transforms::BatchTransform< InputBatch, OutputBatch >:

Public Types
using	InputBatchType = InputBatch
using	OutputBatchType = OutputBatch

Public Member Functions
virtual OutputBatch	apply_batch (InputBatch input_batch)=0
	Applies the transformation to the given input_batch.

Detailed Description

template<typename InputBatch, typename OutputBatch>
class torch::data::transforms::BatchTransform< InputBatch, OutputBatch >

A transformation of a batch to a new batch.

Definition at line 14 of file base.h.

---

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

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