Tensorflow: Access index of variable containing an array

Question

I need to save some values to specific places in a tensorflow array:

import tensorflow as tf
import numpy as np

AVG = tf.Variable([0, 0, 0, 0, 0], name='data')

for i in range(5): 
   data = np.random.randint(1000, size=10000)
   AVG += np.average(data)

I need to make it average each iteration in different places of the AVG variable. Is this doable ?

keveman · Accepted Answer · 2016-10-20 21:36:42Z

1

You can use tf.scatter_add. Here is a complete working program :

import tensorflow as tf
import numpy as np

AVG = tf.Variable([0, 0, 0, 0, 0], name='data')

for i in range(5):
  data = np.random.randint(1000, size=10000)
  AVG = tf.scatter_add(AVG, [i], [np.average(data).astype('int')])

sess = tf.InteractiveSession()
sess.run(tf.initialize_all_variables())
print(AVG.eval())

answered Oct 20, 2016 at 21:36

keveman

8,4851 gold badge41 silver badges47 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Tensorflow: Access index of variable containing an array

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related