如何在theano中组合两个张量

Question

我想知道如何将两个变量组合起来，类似于 python 中的追加？例如，我们有两个变量（输入数据后）：

x: 尺寸为 1*3 y：大小为 1*3

现在我想要一个变量，它将 x 和 y 组合成 1*3*2 的大小

谢谢，

Answer 1

可以使用 theano.tensor.stack 来实现这一点。这是一个工作示例：

import theano
import theano.tensor as tt

x = tt.matrix()
y = tt.matrix()
z = tt.stack([x, y], axis=2)
f = theano.function([x, y], z)
print f([[1, 2, 3]], [[4, 5, 6]])

打印

[[[ 1.  4.]
  [ 2.  5.]
  [ 3.  6.]]]

如何在theano中组合两个张量

How to combine two tensor in theano

python

theano