2016-11-01 26 views
5
import tensorflow as tf 
x = [[1,2,3],[4,5,6]] 
y = [0,1] 
z = [1,2] 
x = tf.constant(x) 
y = tf.constant(y) 
z = tf.constant(z) 
m = x[y,z] 

olduğunu gerekir. Tensorflow kullanarak sonucu nasıl alırım?tensorflow ValueError: Şekil sıralaması 1 olabilir ama ben Theano veya Numpy tarafından sonucu alabilirsiniz <code>m = [2,6]</code></p> <p>bekliyoruz ne rütbe 2

+0

Bence bu bir tensorflow problemi değil, python'daki etkinlik çalışmıyor. eğer x tanımlanırsanız bir liste ve istediğiniz gibi bir listeye erişemezsiniz. X'i numpy.array olarak tanımlayın ([[1,2,3], [4,5,6]]) ve işe yarayacaktır. – Mquinteiro

+0

Sayısal diziyle aynı, yorumum yanlış. – Mquinteiro

cevap

5

Bu yardımcı tf.gather_nd

slices = tf.gather_nd(x, [y, z]) 

Umut kullanmak istersiniz.

+0

Teşekkür ederim, tf.pack() ve tf.gather_nd() yöntemini kullanın. –

+0

Ama sonra gradient of gather_nd hatası alıyorum ve uygulanmadı. T_T, her neyse, teşekkürler –

İlgili konular