theano tensor.stack函数

import theano
import numpy as np
import theano.tensor as T
ones = theano.shared(np.float32([[1,2,3],[4,5,6],[7,8,9]]))

print(ones.get_value())

result = T.stack([ones,ones],axis=0)  #（第一维度拼接两个ones张量）

print(result.eval())

result = T.stack([ones,ones],axis=1)  #（第2维度拼接两个ones张量）

print(result.eval())

result = T.stack([ones,ones],axis=2)  #（第3维度拼接两个ones张量）

print(result.eval())

运行结果：

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

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

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

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