从几个向量制作矩阵

时间:2015-11-21 06:54:19

标签: theano

是否有可能用theano中的几个向量制作矩阵?

像:

vector1, vector2, vector3 = theano.tensor.vector()
Matrix = [vector1, vector2, vector3]

类似于numpy操作:

Matrix = numpy.asarray([vector1, vector 2, vector3])

1 个答案:

答案 0 :(得分:3)

您可以使用theano.tensor.stack

这是一个有效的例子:

import theano
import theano.tensor as tt

vector1, vector2, vector3 = tt.vectors(3)
matrix = tt.stack(vector1, vector2, vector3)
f = theano.function([vector1, vector2, vector3], matrix)
print f([1, 2, 3], [4, 5, 6], [7, 8, 9])

其中print

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