theano dimshuffle 实例

import theanoimport numpy as npones = theano.shared(np.float32([[1,2,3],[4,5,6],[7,8,9]]))temp = ones.dimshuffle([0,1,'x'])print(temp.eval())temp = ones.dimshuffle([0,1])print(temp.eval())temp = ones.dimshuffle([1,0])print(temp.eval())

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

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

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