Update segmento de tensor en tensorflow
Quiero actualizar una porción de un tensor con 3 dimensiones. Siguiendo Cómo hacer una asignación de corte en Tensorflow Haría algo como
import tensorflow as tf
with tf.Session() as sess:
init_val = tf.Variable(tf.zeros((2, 3, 3)))
indices = tf.constant([[0, 0, 0], [0, 0, 1], [0, 1, 0], [0, 1, 1]])
update = tf.scatter_nd_add(init_val, indices, tf.ones(4))
init = tf.global_variables_initializer()
sess.run(init)
print(sess.run(update))
Esto funciona, pero dado que mi problema real es más complejo, me gustaría generar el conjunto de índices de alguna manera automáticamente definiendo el comienzo y el tamaño de la porción, como si usaratf.slice(...)
. ¿Tienes alguna idea? ¡Gracias por adelantado
Estoy usando TensorFlow 1.12, que actualmente es la versión más reciente.