Как можно использовать индекс в tenorflow? - PullRequest
Новая
       8

Как можно использовать индекс в tenorflow?

0 голосов
/ 30 августа 2018

У меня три 2-дим тензора на тренировке. Я сделал так, как показано ниже, но я получаю ошибку

Невозможно преобразовать неизвестное измерение в тензор:? 

mu = params[:, :self.num_mixture]
sigma = params[:, self.num_mixture:self.num_mixture * 2]
alpha = params[:, self.num_mixture * 2:self.num_mixture * 3]
max_alpha = tf.cast(tf.argmax(alpha, axis=1 ), 'int32')

mu = tf.gather_nd(mu,tf.stack([tf.range(max_alpha.shape[0])[...,tf.newaxis], mu[...,tf.newaxis]], axis=2))
sigma = tf.gather_nd(sigma,tf.stack([tf.range(max_alpha.shape[0])[...,tf.newaxis], sigma[...,tf.newaxis]], axis=2))

Как я могу это исправить?

...