Mandelbrot Set分型动画 使用Taichi编程

import taichi as ti

ti.init(arch=ti.gpu)

n = 320
pixels = ti.var(dt=ti.f32, shape=(n * 2, n))

@ti.func
def Complex_sqr(z):
    return ti.Vector([z[0]**2 - z[1]**2, z