Update.
[pytorch.git] / mandelbrot.py
1 #!/usr/bin/env python
2
3 import torch, torchvision
4
5 n = 1280
6
7 cr = torch.linspace(-2, 1, n).view(1, -1)
8 ci = torch.linspace(-1.5, 1.5, n).view(-1, 1)
9
10 zr = torch.zeros(n, n)
11 zi = torch.zeros(n, n)
12
13 for k in range(100):
14     zr, zi = zr**2 - zi**2 + cr, 2 * zr * zi + ci
15
16 torchvision.utils.save_image(1 - (1 - zr**2 + zi**2).sign(), "mandelbrot.png")