def METHOD_NAME(iters_per_epoch: int, epoch: int, p: float, sigma: int, tau: int, layer: torch.nn.Module,
                 module_index: int):
    """Defines a replacement policy from a `torch.nn.Dropout` to a 'GyroDropout`"""
    return GyroDropoutLayer(iters_per_epoch, epoch, p, sigma, tau)