Theano: Warum schlägt die Indexierung in diesem Fall fehl?

Question

Ich versuche, das Maximum eines Vektors mit einem booleschen Wert zu erhalten.

Mit Numpy:

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

Aber mit 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' 

Warum passiert das das? Ist das eine subtile Nuance, die ich vermisse?

Answer 1

Sie verwenden eine Version von Theano, die zu alt ist. Tatsächlich ist tensor_var.nonzero() in keiner freigegebenen Version. Sie müssen auf die Entwicklungsversion aktualisieren.

Mit der Entwicklung Version Ich habe dies:

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

Dies liegt daran, dass Sie Klammern in Ihrer Linie fehlen. Hier ist die gute Linie:

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

Aber wir haben immer noch unerwartetes Ergebnis! Das Problem ist, dass Theano Bool nicht unterstützt. Doing ~ auf int8 macht das bitweise Invertieren von 8 Bits, nicht 1 Bit. Es macht dieses Ergebnis:

>>> (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) 

Sie können die ~ mit diesem entfernen:

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