python - 如何在没有 for 循环的情况下提取 theano 张量中的成对对角线？

我有一个 4 维张量“a”，尺寸为 m x n x n x k。我想获得以下张量“b”:使用 python 我有:

 b = [a[:,0,0,3:-3]]
 for i in xrange(1,n):
    b=numpy.concatenate((b,[a[:,i,i,3:-3]]))

如何在不使用 for 循环的情况下将其转换为 theano 代码？ n 约为 500，因此如果我只在 theano 中执行 for 循环，它会崩溃。

最佳答案

你可以试试这个:

i = theano.shared(numpy.asarray(range(n)))
b = a[:,i,i,3:-3]

Theano基本继承了Numpy's array indexing .

关于python - 如何在没有 for 循环的情况下提取 theano 张量中的成对对角线？，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/36491298/