Tensorflow：配列

Question

を含む変数のアクセス指数Iはtensorflow配列内の特定の場所にいくつかの値を保存する必要があります。

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)  

私はそれがAVG変数の異なる場所で各反復を平均化しておく必要があります。これは実行可能ですか？

Answer 1

tf.scatter_addを使用できます。ここに完全な作業プログラムがあります：

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())