Tensorflow: l'espace réservé après tf.split() n'est pas accessible via une autre variable d'index d'espace réservé?

Question

J'ai une variable tensorflow qui est divisée en utilisant tf.split.

Theta = tf.Variable(tf.random_normal((R,s), dtype=tf.float64))
Theta_s = tf.split(Theta, ysplit, 1)

ysplit est une liste contenant la longueur de la fente le long de l'axe 1. Maintenant Theta_s[i] est une matrice de dimension (R x ysplit[i]). Je dois accéder Theta_s à travers une autre variable d'index de placeholder. À l'heure actuelle, je ne suis pas en mesure de le faire depuis tf.split() retourne la liste et je reçois cette erreur:

TypeError: list indices must be integers or slices, not Tensor

Y at-il un bon moyen de déclarer la variable à répondre à l'objectif?

Answer 1

Vous pouvez utiliser un tf.TensorArray pour effectuer l'indexation dynamique avec une tf.Tensor:

Theta_s = tf.split(Theta, ysplit, 1) 

array = tf.TensorArray(tf.float64, size=len(Theta_s), clear_after_read=False) 

for i, t in enumerate(Theta_s): 
    array = array.write(i, t) 

placeholder_index = tf.placeholder(tf.int32, shape=[]) 

Theta_s_i = array.read(placeholder_index)