在Theano中向张量中添加一行-Java 学习之路

我是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但是无法达到预期的效果 .

非常感谢您的帮助！本

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.]])

回复于 2024-05-01T20:08:20+08:00

在Theano中向张量中添加一行

1 回答

相关问题