Dropout

class complextorch.nn.modules.dropout.CVDropout(p: float = 0.5, inplace: bool = False)

Complex-Valued Dropout Layer

Applies PyTorch Droput to real and imaginary parts separately.

Implements the following operation:

\[G(\mathbf{z}) = \texttt{Dropout}(\mathbf{x}) + j \texttt{Dropout}(\mathbf{y}),\]

where \(\mathbf{z} = \mathbf{x} + j\mathbf{y}\)

forward(input: CVTensor) CVTensor

Performs complex-valued dropout on the input tensor

Parameters:

x (CVTensor) – input tensor

Returns:

\(\texttt{Dropout}(\mathbf{x}) + j \texttt{Dropout}(\mathbf{y})\)

Return type:

CVTensor