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")