theano - 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 并使用索引向量对其进行索引。

Answer 1

你可以做：

A[v]

它会做你想做的事。