def METHOD_NAME() -> None:
    torch.set_default_dtype(torch.float64)
    data, labels = tetris()
    data = make_batch(data)
    f = Network()
    pred = f(data)
    loss = (pred - labels).pow(2).sum()
    loss.backward()
    rotated_data, _ = tetris()
    rotated_data = make_batch(rotated_data)
    error = f(rotated_data) - f(data)
    assert error.abs().max() < 1e-10