Theano：スキャンの反復インデックスを呼び出された関数に渡す

Question

スキャンの反復インデックスをスキャンから呼び出す関数に渡すことはできますか？例えばのために -

def step(x,i): 
    # i is the current scan index. Use it for some conditional expressions 

for i in range(0,10): 
    step(x,i) 

私はtheanoを使用して、同様の何かをしたいです。すべての手がかりは？

おかげ

Answer 1

これは、例えばhereため、theanoのチュートリアルに示されています。

機能の最初の引数は、scanの引数sequences（提供されている場合）から自動的に取得されます。 たとえば、ベクトルxのすべての要素に対応するインデックスを追加するとします。 次のようなtheano.scanでそれを行うことができます：

import theano 
import theano.tensor as T 
x = T.dvector('x') 
def step(idx, array): 
    return array[idx] + idx 
results, updates = theano.scan(fn=step, 
           sequences=T.arange(x.shape[0]), 
           non_sequences=[x]) 
f = theano.function([x], results) 
f([1., 0., 2.13]) 
# array([ 1. , 1. , 4.13]) 

をので、基本的に：あなたはscanに与える関数の引数の順番を気に、彼らは特定の順序で渡されます。 該当するdocumentation page of scanの注文を正確に見ることができます。