PreprocessingPipeline

class PreprocessingPipeline<I, M, O>(firstOp: Operation<I, M>, secondOp: Operation<M, O>) : Operation<I, O>
Content copied to clipboard

This class is a special type of Operation which is used to build typesafe pipeline of preprocessing operations. Kudos to @juliabeliaeva for the idea.

Constructors

PreprocessingPipeline
Link copied to clipboard
fun <I, M, O> PreprocessingPipeline(firstOp: Operation<I, M>, secondOp: Operation<M, O>)
Content copied to clipboard

Functions

apply
Link copied to clipboard
open override fun apply(input: I): O
Content copied to clipboard

Performs preprocessing operation on the input.

getOutputShape
Link copied to clipboard
open override fun getOutputShape(inputShape: TensorShape): TensorShape
Content copied to clipboard

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