def forward(self, batch: torch.Tensor) -> torch.Tensor:
        """Generate predicted anomaly masks.

        Args:
            batch (torch.Tensor): Concatenated original and reconstructed images of
                shape ``(batch_size, channels*2, height, width)``

        Returns:
            torch.Tensor: Pixel-level class scores for normal and anomalous regions
        """
        act1, act2, act3, act4, act5, act6 = self.encoder_segment(batch)
        return self.decoder_segment(act1, act2, act3, act4, act5, act6)
Generate predicted anomaly masks.

        Args:
            batch (torch.Tensor): Concatenated original and reconstructed images of
                shape ``(batch_size, channels*2, height, width)``

        Returns:
            torch.Tensor: Pixel-level class scores for normal and anomalous regions
        
src/anomalib/models/image/draem/torch_model.py