import torch


def get_pyr_shapes(N, r, start_shape=(25, 25)):
    """Get the shape of the images/noise maps of each scale."""

    # Init variables
    shape = torch.tensor(start_shape)
    shapes = []

    # For every scale n, create the noise map with right scale
    for n in range(N):
        shapes.append(tuple(shape.int().tolist()))
        shape = shape*r
    return shapes