用theano中的向量值索引矩阵答案

【问题标题】：Indexing matrix with values from vector in theano用theano中的向量值索引矩阵
【发布时间】：2016-06-23 00:24:58
【问题描述】：

我正在研究 theano 中的成本函数并遇到以下问题：我有一个矩阵 M（比如[[3,4],[5,6]]），一个向量 v（[0,1]），我想获得 M 中对应于 v 中包含的索引的条目向量（这里是 [3,6]）。有没有命令可以做到这一点？

编辑：我最终做了以下事情：M[T.arange(v.shape[0]), v]，其中 T.arange(v.shape[0]) 只是一个向量，其中的条目是整数v.M 的长度范围需要用两个向量的列表来索引

【问题讨论】：

【解决方案1】：

这应该可行。

VecOut = [M[i][j] for i,j in enumerate(V)]

enumerate 将索引和向量值返回给 i（行索引）和 j（向量分量的值），可用于从矩阵中获取对应的值。

示例：

【讨论】：