Operation

interface Operation<I, O>

Interface for preprocessing operations.

Parameters

I

is a type of an input.

O

is a type of an output.

Functions

apply
Link copied to clipboard
abstract fun apply(input: I): O

Performs preprocessing operation on the input.

getOutputShape
Link copied to clipboard
abstract fun getOutputShape(inputShape: TensorShape): TensorShape

Returns shape of the output of the operation having input of shape inputShape.

Inheritors

Identity
Link copied to clipboard
PreprocessingPipeline
Link copied to clipboard