def forward(self, batch: torch.Tensor) -> torch.Tensor:
        """Reconstructs an input batch of images.

        Args:
            batch (torch.Tensor): Batch of input images.

        Returns:
            Reconstructed images.
        """
        b1, b2, b3, b4 = self.encoder(batch)
        return self.decoder(b1, b2, b3, b4)
Reconstructs an input batch of images.

        Args:
            batch (torch.Tensor): Batch of input images.

        Returns:
            Reconstructed images.
        
src/anomalib/models/image/dsr/torch_model.py