Theano'da 0/1 altçaplama nasıl yapılır?

Question

Amaç, bir dizi dizinin alt kümesini başka bir dizide sağlanan değerler aracılığıyla elde etmektir.

import theano 
import theano.tensor as T 

a = T.vector('X', dtype='int64') 
b = T.vector('Y', dtype='int64') 
c = a[b] 
g = function([a,b],c) 

x = np.array([5,3,2,3,4,6], dtype=int) 
y = np.array([0,0,1,0,0,1], dtype=int) 
print g(x,y) 

Bu

[5 5 3 5 5 3] 

yerine

[2 6] 

Nasıl beklenen sonucu alabilirim yazdırır?

Answer 1

nonzero() işlevini kullanmayı deneyin. senin durumunda

Örnek:

import theano 
import theano.tensor as T 

a = T.vector('X', dtype='int64') 
b = T.vector('Y', dtype='int64') 
c = a[b.nonzero()] 
g = function([a,b],c) 

x = np.array([5,3,2,3,4,6], dtype=int) 
y = np.array([0,0,1,0,0,1], dtype=int) 
print g(x,y) 

o