TensorFlow配列の要素代入

Question

TensorFlowの配列の要素に値を代入する方法を知りたいです。
例えば、x = tf.Variable(tf.zeros((4,3)))のような配列の各要素に対して、次のように1を代入しようとするとエラーになります。

import tensorflow as tf

x = tf.Variable(tf.zeros((4, 3)))
for i in range(4):
for j in range(3):
x = x[i, j].assign(1)

1度目の代入の後に、xがVariableからstrided_sliceに変化するためだと思われます。
(要素への代入はVariableのみに対応しているので、2度目以降の代入は不可能。)

何か良い方法がございましたら、お願いします。

Answer 1

何か良い方法がございましたら、お願いします。

本来の目的（最終的にやりたいこと）がよく分からないので良い方法かも分かりませんが…
以下のようにnumpy.arrayを生成してassignすることができます。

import tensorflow as tf
import numpy as np

def func(x):
    a = np.zeros(x.shape)
    for i in range(x.shape[0]):
        for j in range(x.shape[1]):
            a[i,j] = 10*i+j
    return a

x = tf.Variable(tf.zeros((4,3)))
op = tf.assign(x, func(x))

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    ret = sess.run(op)
    print(type(ret)) # <class 'numpy.ndarray'>
    print(ret)
    """
    [[ 0.  1.  2.]
     [10. 11. 12.]
     [20. 21. 22.]
     [30. 31. 32.]]
    """

    print(type(x)) # <class 'tensorflow.python.ops.variables.Variable'>
    print(x.eval())
    """
    [[ 0.  1.  2.]
     [10. 11. 12.]
     [20. 21. 22.]
     [30. 31. 32.]]
    """