Method __call__

monai/visualize/gradient_based.py:173–175

(self, x: torch.Tensor, index: torch.Tensor | int | None = None, **kwargs: Any)

    """

    def __call__(self, x: torch.Tensor, index: torch.Tensor | int | None = None, **kwargs: Any) -> torch.Tensor:
        with replace_modules_temp(self.model, "relu", _GradReLU(), strict_match=False):
            return super().__call__(x, index, **kwargs)