def forward(self, x):
        """
        Forward pass for the SELayer.

        The forward method applies the squeeze-and-excitation operation on the input tensor `x`.
        It computes the channel-wise attention weights for both the real and imaginary parts 
        of the input.

        Args:
            x (torch.Tensor): Input tensor of shape (B, C, D, H, W), where:
                              B - batch size,
                              C - number of channels,
                              D - depth,
                              H - height,
                              W - width.

        Returns:
            torch.Tensor: Output tensor after applying channel-wise attention, 
                          same shape as input `x`.
        """
        # Extract the batch size and number of channels
        b, c, _, _, _ = x.size()

        # Compute the squeeze operation for the real part
        x_r = self.avg_pool(x[:, :, :, :, 0]).view(b, c)  # Global average pooling
        # Compute the squeeze operation for the imaginary part
        x_i = self.avg_pool(x[:, :, :, :, 1]).view(b, c)  # Global average pooling

        # Calculate channel-wise attention for the real part
        y_r = self.fc_r(x_r).view(b, c, 1, 1, 1) - self.fc_i(x_i).view(b, c, 1, 1, 1)
        # Calculate channel-wise attention for the imaginary part
        y_i = self.fc_r(x_i).view(b, c, 1, 1, 1) + self.fc_i(x_r).view(b, c, 1, 1, 1)

        # Concatenate real and imaginary attention weights along the channel dimension
        y = torch.cat([y_r, y_i], 4)

        # Scale the input features by the attention weights
        return x * y

        Forward pass for the SELayer.

        The forward method applies the squeeze-and-excitation operation on the input tensor `x`.
        It computes the channel-wise attention weights for both the real and imaginary parts 
        of the input.

        Args:
            x (torch.Tensor): Input tensor of shape (B, C, D, H, W), where:
                              B - batch size,
                              C - number of channels,
                              D - depth,
                              H - height,
                              W - width.

        Returns:
            torch.Tensor: Output tensor after applying channel-wise attention, 
                          same shape as input `x`.
        
clearvoice/clearvoice/models/frcrn_se/se_layer.py