def test_scatter_nd(): indices = tf.constant([[3]]) updates = tf.constant([[9,10,11,12]]) shape = tf.constant([8,4]) scatter = tf.scatter_nd(indices, updates, shape) return scatter