Class _GradReLU

monai/visualize/gradient_based.py:48–55

A customized ReLU with the backward pass imputed for guided backpropagation (https://arxiv.org/abs/1412.6806).

class _GradReLU(torch.nn.Module):
    """
    A customized ReLU with the backward pass imputed for guided backpropagation (https://arxiv.org/abs/1412.6806).
    """

    def forward(self, x: torch.Tensor) -> torch.Tensor:
        out: torch.Tensor = _AutoGradReLU.apply(x)
        return out


class VanillaGrad: