[PYTHON/TENSORFLOW] 망델브로 집합 출력하기

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

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

zNDArray = xNDArray + 1j * yNDArray

cTensor = tf.constant(zNDArray.astype(np.complex64))

zsVariable = tf.Variable(cTensor)
nsVariable = tf.Variable(tf.zeros_like(cTensor, tf.float32))

session = tf.InteractiveSession()

tf.global_variables_initializer().run()

zsTensor = zsVariable * zsVariable + cTensor

notDivergedTensor = tf.abs(zsTensor) < 4

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

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

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

 

728x90

 

[결과]