Theano中的索引

Question

我该如何使用Theano将矩阵按照向量索引？ 更具体地说，假设：

• v 是 theano.tensor.vector 类型 (例如 [0,2])
• A 是 theano.tensor.matrix 类型 (例如 [[1,0,0], [0,1,0], [0,0,1]])

期望的结果是[[1,0,0], [0,0,1]]。 提到一下我的目标是将索引列表转换为仅有一个热点的行向量矩阵，其中索引指示了热点所在位置。我的初始尝试是让A = theano.tensor.eye 并使用向量索引它。

- John Jaques

Answer 1

你可以做以下事情：

A[v]

它将按照你的要求执行。

- nouiz

---

我已经尝试过了，但是出现了错误信息。我定义了v = theano.tensor.vector()、A = theano.tensor.matrix()和indexed = A[v]。错误信息如下：TypeError: index must be integers。 - John Jaques

列表v必须是整数。您可以像这样转换为int：theano.tensor.cast(v，'int64')。 - nouiz

v = theano.tensor.lvector() 给你一个64位整数向量，无需转换类型。在此处查看类型列表：http://deeplearning.net/software/theano/library/tensor/basic.html#all-fully-typed-constructors - JStrahl