Question

我是theano的新手，我正试图找出如何在矩阵张量中添加一行1，其中输入张量的第一个形状会略有不同，例如：（50000,784），（10000,784）使用MNIST。举一个笨拙的例子;这就是我想在theano中实现的，其中数组将是张量：

array = np.zeros((2,2))
array = np.hstack((np.ones((2,1)), array))

我查看过shape_padleft以及pad_right和padaxis，但无法达到预期效果。

非常感谢您的帮助！本

Answer 1

所以回答我自己的问题;）我希望这可以帮助其他人！

A = T.matrix('A')
B = T.ones_like(A[:,0])
C = T.concatenate([T.shape_padright(B), A], axis=1)

C.eval({A:np.zeros((2,2))})

>> array([[1., 0., 0.],
          [1., 0., 0.]])

在Theano中向张量中添加一行

1 个答案: