torch::data::transforms::BatchLambda< Input, Output > Class Template Reference

A BatchTransform that applies a user-provided functor to a batch. More...

#include <lambda.h>

Inheritance diagram for torch::data::transforms::BatchLambda< Input, Output >:

Public Types
using	FunctionType = std::function< OutputBatchType(InputBatchType)>
Public Types inherited from torch::data::transforms::BatchTransform< Input, Output >
using	InputBatchType = Input
using	OutputBatchType = Output

Public Member Functions
	BatchLambda (FunctionType function)
	Constructs the BatchLambda from the given function object.
OutputBatchType	apply_batch (InputBatchType input_batch) override
	Applies the user-provided function object to the input_batch.

Detailed Description

template<typename Input, typename Output = Input>
class torch::data::transforms::BatchLambda< Input, Output >

A BatchTransform that applies a user-provided functor to a batch.

Definition at line 15 of file lambda.h.

---

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

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