quantax.nn.Scale

class quantax.nn.Scale

Bases: NoGradLayer

Rescale the input \(f(x) = x * \mathrm{scale}\)

__init__(scale: float)

Attributes

scale