Theano: Tại sao việc lập chỉ mục thất bại trong trường hợp này?

Question

Tôi đang cố gắng tối đa hóa một vectơ cho một giá trị boolean.

Với NumPy:

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

Nhưng với 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' 

Tại sao điều này xảy ra? Đây có phải là một sắc thái tinh tế mà tôi đang thiếu?

Answer 1

Bạn đang sử dụng phiên bản Theano quá cũ. Trong thực tế, tensor_var.nonzero() không có trong bất kỳ phiên bản phát hành nào. Bạn cần cập nhật lên phiên bản phát triển.

Với phiên bản phát triển Tôi có điều này:

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

Điều này là do bạn đang thiếu ngoặc phù hợp của bạn. Đây là dòng tốt:

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

Nhưng chúng tôi vẫn có kết quả bất ngờ! Vấn đề là Theano không hỗ trợ bool. Làm ~ trên int8, đang làm ngược bitwise trên 8 bit, không phải 1 bit. Nó cho kết quả này:

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

Bạn có thể loại bỏ các ~ với điều này:

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