Возврат индекса значения в векторе Anano

Question

Какова процедура в Theano для возврата индекса определенного значения в вектор? В NumPy это будет numpy.where(my_array==x). Theano's Tensor.where является оператором switch.

Answer 1

Существует 2 поведения numpy.where (condition, [x, y]). Theano всегда поддерживает, что вы предоставляете 3 параметра where(). Как указано в NumPy doc [1], numpy.where (cond) эквивалентно ненулевому().

Вы можете сделать это, как это в Феано:

import theano 
import numpy as np 
v = np.arange(10) 
var = theano.tensor.vector() 
out = theano.tensor.eq(var, 2).nonzero()[0] 
print out.eval({var: v}) 

Проверка линии 5. NumPy ненулевую() возвращает кортеж. Теано делает то же самое. Существует один вектор в этом наборе для измерений на входе ненулевого().

[1] http://docs.scipy.org/doc/numpy/reference/generated/numpy.where.html