Noise¶
-
class
torchelie.nn.Noise(ch: int, inplace: bool = False, bias: bool = False)¶ Add gaussian noise to the input, with a per channel or global learnable std.
- Parameters
ch (int) – number of input channels for a different std on each channel, or 1
-
forward(x: torch.Tensor, z: Optional[torch.Tensor] = None) → torch.Tensor¶
-
training: bool¶