Управление матричными элементами в тензорном потоке
Как я могу сделать следующее в tenorflow?
mat = [4,2,6,2,3] #
mat[2] = 0 # simple zero the 3rd element
Я не могу использовать скобки [], потому что он работает только с константами, а не с переменными. Я также не могу использовать функцию слайса, потому что она возвращает тензор, и вы не можете назначить тензор.
import tensorflow as tf
sess = tf.Session()
var1 = tf.Variable(initial_value=[2, 5, -4, 0])
assignZerosOP = (var1[2] = 0) # < ------ This is what I want to do
sess.run(tf.initialize_all_variables())
print sess.run(var1)
sess.run(assignZerosOP)
print sess.run(var1)
Будет печатать[2, 5, -4, 0]
[2, 5, 0, 0])