[PYTHON/TENSORFLOW] 쥘리아 집합 출력하기

import matplotlib.pyplot as pp
import numpy as np
import tensorflow as tf

yNDArray, xNDArray = np.mgrid[-2:2:0.005, -2:2:0.005]

zNDArray = xNDArray + 1j * yNDArray

zTensor = tf.constant(zNDArray.astype("complex64"))

zsVariable = tf.Variable(zTensor)
nsVariable = tf.Variable(tf.zeros_like(zTensor, "float32"))

tf.InteractiveSession()

tf.global_variables_initializer().run()

cComplex = complex(0.0, 0.75)

zsTensor = zsVariable * zsVariable - cComplex

notDivergedTensor = tf.abs(zsTensor) < 4

stepOperation = tf.group(zsVariable.assign(zsTensor), nsVariable.assign_add(tf.cast(notDivergedTensor, "float32")))

for i in range(200):
    stepOperation.run()

pp.imshow(nsVariable.eval())
pp.show()

 

728x90

 

[결과]