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.

Respuestas a la pregunta(1)

Su respuesta a la pregunta