Skip to content

rescaling

Rescaling Layer API

This module provides classes to build rescaling layers.

Classes:

Classes

Rescaling1D 

Rescaling1D(scale: float, **kwargs)

Rescale the input samples.

Parameters:

  • scale (float) –

    The scaling factor.

Source code in neuralspot_edge/layers/preprocessing/rescaling.py 
21
22
23
24
25
26
27
28
def __init__(self, scale: float, **kwargs):
    """Rescale the input samples.

    Args:
        scale (float): The scaling factor.
    """
    super().__init__(**kwargs)
    self.scale = scale

Functions

augment_samples 
augment_samples(inputs) -> keras.KerasTensor

Rescale a batch of samples during training.

Source code in neuralspot_edge/layers/preprocessing/rescaling.py 
30
31
32
33
def augment_samples(self, inputs) -> keras.KerasTensor:
    """Rescale a batch of samples during training."""
    samples = inputs[self.SAMPLES]
    return samples * self.scale
compute_output_shape 
compute_output_shape(input_shape, *args, **kwargs)

Compute output shape.

Source code in neuralspot_edge/layers/preprocessing/rescaling.py 
35
36
37
def compute_output_shape(self, input_shape, *args, **kwargs):
    """Compute output shape."""
    return input_shape
get_config 
get_config()

Serialize the configuration.

Source code in neuralspot_edge/layers/preprocessing/rescaling.py 
39
40
41
42
43
def get_config(self):
    """Serialize the configuration."""
    config = super().get_config()
    config.update(scale=self.scale)
    return config

Rescaling2D 

Rescaling2D(scale: float, **kwargs)

Rescale the input samples.

Parameters:

  • scale (float) –

    The scaling factor.

Source code in neuralspot_edge/layers/preprocessing/rescaling.py 
50
51
52
53
54
55
56
57
def __init__(self, scale: float, **kwargs):
    """Rescale the input samples.

    Args:
        scale (float): The scaling factor.
    """
    super().__init__(**kwargs)
    self.scale = scale

Functions

augment_samples 
augment_samples(inputs) -> keras.KerasTensor

Rescale a batch of samples during training.

Source code in neuralspot_edge/layers/preprocessing/rescaling.py 
59
60
61
62
def augment_samples(self, inputs) -> keras.KerasTensor:
    """Rescale a batch of samples during training."""
    samples = inputs[self.SAMPLES]
    return samples * self.scale
compute_output_shape 
compute_output_shape(input_shape, *args, **kwargs)

Compute output shape.

Source code in neuralspot_edge/layers/preprocessing/rescaling.py 
64
65
66
def compute_output_shape(self, input_shape, *args, **kwargs):
    """Compute output shape."""
    return input_shape
get_config 
get_config()

Serialize the configuration.

Source code in neuralspot_edge/layers/preprocessing/rescaling.py 
68
69
70
71
72
def get_config(self):
    """Serialize the configuration."""
    config = super().get_config()
    config.update(scale=self.scale)
    return config