theano scan arange shape 实例

import theanoimport numpy as npimport theano.tensor as Tones1 = theano.shared(np.float32([[1,2,3],[4,5,6],[7,8,9]]))ones2 = theano.shared(np.float32([[9,8,7],[6,5,4],[3,2,1]]))print(ones1.eval())print(T.shape(ones1).eval())print(T.arange(T.shape(ones1)[0]).eval())outputs, updates = theano.scan(lambda ones1,ones2 : ones2 + 1,                               sequences = [ones1,T.arange(T.shape(ones1)[0])])print(outputs.eval())

结果：

[[ 1. 2. 3.]
[ 4. 5. 6.]
[ 7. 8. 9.]]

[3 3]

[0 1 2]

[1 2 3]