TensorFlow配列の要素代入

Question

TensorFlowの要素に値を代入することは可能ですか？
具体的には以下のような配列の要素に代入を行おうとするとエラーが現れます。

x = tf.Variable(tf.zeros(4, 3))
x[0][0] = 1

---

TypeError Traceback (most recent call last)
<ipython-input-4-ad6d340277e3> in <module>()
1 x = tf.Variable(tf.zeros(4, 3))
----> 2 x[0][0] = 1

TypeError: 'Tensor' object does not support item assignment

よろしくお願いします。

Answer 1

まず、4x3の行列を作成するにはtf.zeros(4,3)ではなくtf.zeros((4,3))と書きます。
また、代入するにはassignを使います。

import tensorflow as tf

x = tf.Variable(tf.zeros((4,3)))
x = x[0,0].assign(1)

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    ret = sess.run(x)
    print(ret)
"""
[[1. 0. 0.]
 [0. 0. 0.]
 [0. 0. 0.]
 [0. 0. 0.]]
"""

Comments

[opeco17]

ご丁寧な回答有り難うございます。
xがplaceholderの場合、x = x[0,0].assign(1)を使えないのでしょうか？
試してみた所、エラーが出てしまいます。

x = tf.placeholder(tf.float32, shape = [4, 3])
x = x[0, 0].assign(1)
ValueError: Sliced assignment is only supported for variables

[can110]

エラーメッセージに書かれているとおり、placeholderでは使えないようですね。

[opeco17]

承知しました。
ありがとうございます。