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]

它会做你想做的。