TensorPreprocessing

class TensorPreprocessing
Content copied to clipboard

Represents the tensor preprocessing stage of the Preprocessing. Consists of the operations implementing Preprocessor which are applied to the tensor one by one.

Supported operations include:

See also

Constructors

TensorPreprocessing
Link copied to clipboard
fun TensorPreprocessing()
Content copied to clipboard

Functions

addOperation
Link copied to clipboard
fun addOperation(operation: Preprocessor)
Content copied to clipboard

Adds an operation to the operations.

Extensions

normalize
Link copied to clipboard
fun TensorPreprocessing.normalize(block: Normalizing.() -> Unit)
Content copied to clipboard

Applies Normalizing preprocessor to the tensor to normalize it with given mean and std values.

rescale
Link copied to clipboard
fun TensorPreprocessing.rescale(block: Rescaling.() -> Unit)
Content copied to clipboard

Applies Rescaling preprocessor to the tensor to scale each value by a given coefficient.