)
    x = torch.flatten(x, 1)
    x = self.fc(x)
    &#39;&#39;&#39;

    rel, irrel = cd_generic(mods[1:5], rel, irrel)

    lay_nums = [5, 18, 34, 50]
    for lay_num in lay_nums:
        for basic_block in mods[lay_num]:
            rel, irrel = propagate_basic_block(rel, irrel, basic_block)

    # final things after BasicBlocks
    rel, irrel = cd_generic(mods[-2:], rel, irrel)
    return rel, irrel


def cd_propagate_mnist(relevant, irrelevant, model):
    &#39;&#39;&#39;Propagate a specific mnist arch