使用来自theano中的向量的值来索引矩阵

Question

我正在研究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长度的范围需要用两个向量的列表索引

Answer 1

这应该有效。

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

枚举返回索引和向量值为i（行索引）和j（向量组件的值），可用于从矩阵中获取相应的值。

示例：