Update.

[pytorch.git] / mandelbrot.py

diff --git a/mandelbrot.py b/mandelbrot.py
index fa522eb..c284eb0 100755 (executable)
--- a/mandelbrot.py
+++ b/mandelbrot.py
@@ -13,4 +13,4 @@ zi = torch.zeros(n, n)
 for k in range(100):
     zr, zi = zr**2 - zi**2 + cr, 2 * zr * zi + ci
 
-torchvision.utils.save_image(1-(1-zr**2 + zi**2).sign(), 'mandelbrot.png')
+torchvision.utils.save_image(1 - (1 - zr**2 + zi**2).sign(), "mandelbrot.png")