Манипулирование матричными элементами в тензорном потоке

Question

Как я могу сделать следующее в тензорном потоке?

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

Answer 1

Вы не можете изменить тензор - но, как вы отметили, вы можете изменить переменную.

Есть три модели, которые вы могли бы использовать, чтобы сделать то, что вы хотите:

(а) Использование tf.scatter_update непосредственно тыкать в части переменной, которую вы хотите изменить.

import tensorflow as tf 

a = tf.Variable(initial_value=[2, 5, -4, 0]) 
b = tf.scatter_update(a, [1], [9]) 
init = tf.initialize_all_variables() 

with tf.Session() as s: 
    s.run(init) 
    print s.run(a) 
    print s.run(b) 
    print s.run(a) 

[2 5 -4 0]

[2 9 -4 0]

[2 9 -4 0]

(б) Создать два tf.slice() s тензора, за исключением элемента, который вы хотите изменить, а затем tf.concat(0, [a, 0, b]) их обратно вместе.

(с) Создание b = tf.zeros_like(a), а затем использовать tf.select() выбрать, какие элементы из a вы хотите, и какие нули из b, что вы хотите.

Я включил (b) и (c), потому что они работают с нормальными тензорами, а не только с переменными.