Mandelbrot generation

Mandelbrot

m = train_X.shape[1]
n = train_X.shape[2]

def mandelbrot(height, width, x_from=0, x_to=0.1, y_from=0, y_to=0.1, max_iterations=100):
    x = np.linspace(x_from, x_to, width).reshape((1, width))
    y = np.linspace(y_from, y_to, height).reshape((height, 1))
    c = x + 1j * y
    return c
C=mandelbrot(1,28*28).reshape(28,28)