spyrit.core.recon.Denoise_layer.forward

Denoise_layer.forward(inputs: tensor) tensor[source]

Applies a transformation to the incoming data: \(y = A^2/(A^2+x)\).

\(x\) is the input tensor (see inputs) and \(A\) is the standard deviation prior (see self.weight).

Args:

inputs (torch.tensor): input tensor \(x\) of shape \((N, *, in\_features)\)

Returns:

torch.tensor: The transformed data \(y\) of shape \((N, in\_features)\)

Shape: