spyrit.core.prep.Rescale.forward

Rescale.forward(y: tensor) → tensor

Rescale the tensor by dividing it by \(\alpha\).

Args:

y (torch.tensor): Input tensor with arbitrary shape.

Returns:

torch.tensor: Rescaled tensor with the same shape as the input.