返回an中值的索引

Question

返回向量中特定值的索引的步骤是什么？在NumPy中，这将是numpy.where(my_array==x)。Theano的Tensor.where是一个switch语句。在

Answer 1

有两种行为纽比。在哪里（条件，[x，y]）。始终支持您向where（）提供3个参数。正如NumPy doc[1]所说，纽比。在哪里（cond）相当于非零（）。在

您可以在“No”中这样做：

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 nonzero（）返回元组。他们也一样。在nonzero（）的输入中，每个维度的元组中都有一个向量。在

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