返回Theano向量中的值的索引

Question

Theano中用於返回Vector中特定值的索引的過程是什麼？在NumPy中，這將是numpy.where(my_array==x)。 Theano的Tensor.where是一個switch語句。

Answer 1

numpy.where（condition，[x，y]）有2行爲。 Theano始終支持您向where（）提供3個參數。正如在NumPy doc [1]中所說的，numpy.where（cond）等價於非零（）。

你可以做這樣的Theano：

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的非零（）返回一個元組。 Theano也是這樣做的。在nonzero（）的輸入中，每個維度在該元組中有一個向量。

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