Создайте тензор со случайным индексом, равным нулю

Вы можете сделать это с помощью такой функции:

import tensorflow as tf

def zero_rows(x, idx)
    # Turn row indices into a boolean mask
    n = tf.shape(x)[0]
    m = tf.scatter_nd(tf.expand_dims(idx, 1), tf.ones_like(idx, dtype=tf.bool), [n])
    # Select zeros where the indices are or the data elsewhere
    return tf.where(m, tf.zeros_like(x), x)

# Test
with tf.Graph().as_default(), tf.Session() as sess:
    y = zero_rows(x, idx)
    print(sess.run(y, feed_dict = {x: [[1, 2], [3, 4], [5, 6], [7, 8]], idx: [1, 3]}))
    # [[1 2]
    #  [0 0]
    #  [5 6]
    #  [0 0]]