Tensorflow - setting an array element with a sequence.