forループなしでtheanoのテンソルのpairwise対角を抽出するには？

Question

私は、寸法がm×n×n×kの4次元テンソル「a」を持っています。私が持って使ってPython：私は、これはforループを使用せずにtheanoコードに変換することができますどのように

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

私は、次のテンソル「b」を取得したいですか？ nは約500ですので、私がtheanoでforループを実行するとクラッシュします。

Answer 1

あなたはこれを試すことができます。

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

Theanoは基本的にNumpy's array indexingを継承しています。