将python ndarray转换为theano tensor类型变量

时间:2016-12-17 13:18:27

标签: python multidimensional-array theano

我喜欢ndarray:

diag = []
diag.append(np.diag([1,1,0]))
diag.append(np.diag([0,1,1]))
diag
  [array([[1, 0, 0],
   [0, 1, 0],
   [0, 0, 0]]), array([[0, 0, 0],
   [0, 1, 0],
   [0, 0, 1]])]

如何将其转换为float 64类型的Theano tensor变量,matrix?因为我需要像

那样执行点操作
Theano.dot(diag, X) where X is shared variable of type float 64, matrix.

1 个答案:

答案 0 :(得分:4)

只需像这样创建一个SharedVariable

diag_ = theano.shared(np.array(diag).astype("float64"))
theano.dot(diag_, X)

http://deeplearning.net/software/theano/library/compile/shared.html