Trait Transform 

pub trait Transform: Send + Sync {
    // Required method
    fn apply(&self, input: &Tensor<f32>) -> Tensor<f32>;
}

Trait for data transformations.

Required Methods

fn apply(&self, input: &Tensor<f32>) -> Tensor<f32>

Applies the transform to a tensor.

Implementors

impl Transform for Clamp

impl Transform for Compose

impl Transform for DropoutTransform

impl Transform for Flatten

impl Transform for Normalize

impl Transform for RandomCrop

impl Transform for RandomFlip

impl Transform for RandomNoise

impl Transform for Reshape

impl Transform for Scale

impl Transform for ToTensor

impl<F> Transform for Lambda<F>

where F: Fn(&Tensor<f32>) -> Tensor<f32> + Send + Sync,