trw.simple_layers.shift_scale

Module Contents

Classes

ShiftScale

Normalize a tensor with a mean and standard deviation

class trw.simple_layers.shift_scale.ShiftScale(node, mean, standard_deviation)

Bases: trw.simple_layers.simple_layers.SimpleModule

Normalize a tensor with a mean and standard deviation

The output tensor will be (x - mean) / standard_deviation

This layer simplify the preprocessing for the trw.simple_layers package