Theano: Dlaczego indeksowanie kończy się niepowodzeniem?

Question

Próbuję uzyskać maksimum wektora z wartością boolowską.

Z NumPy:

>>> this = np.arange(10) 
>>> this[~(this>=5)].max() 
4 

Ale z Theano:

>>> that = T.arange(10, dtype='int32') 
>>> that[~(that>=5)].max().eval() 
9 
>>> that[~(that>=5).nonzero()].max().eval() 
Traceback (most recent call last): 
    File "<pyshell#146>", line 1, in <module> 
    that[~(that>=5).nonzero()].max().eval() 
AttributeError: 'TensorVariable' object has no attribute 'nonzero' 

Dlaczego tak się dzieje? Czy to subtelny niuans, którego mi brakuje?

Answer 1

Używasz wersji Theano, która jest zbyt stara. W rzeczywistości tensor_var.nonzero() nie występuje w żadnej wydanej wersji. Musisz zaktualizować wersję rozwojową.

W wersji rozwojowej mam to:

>>> that[~(that>=5).nonzero()].max().eval() 
Traceback (most recent call last): 
    File "<stdin>", line 1, in <module> 
TypeError: bad operand type for unary ~: 'tuple' 

To dlatego, że brakuje nawiasu w linii. Oto dobra linia:

>>> that[(~(that>=5)).nonzero()].max().eval() 
array(9, dtype=int32) 

Ale wciąż mamy nieoczekiwany rezultat! Problem polega na tym, że Theano nie wspiera bool. Robiąc ~ na int8, wykonuje bitowe odwrócenie na 8 bitach, a nie na 1 bit. To daje ten wynik:

>>> (that>=5).eval() 
array([0, 0, 0, 0, 0, 1, 1, 1, 1, 1], dtype=int8) 
>>> (~(that>=5)).eval() 
array([-1, -1, -1, -1, -1, -2, -2, -2, -2, -2], dtype=int8) 

można usunąć ~ z tym:

>>> that[(that<5).nonzero()].max().eval() 
array(4, dtype=int32)