如何使用Theano在两个向量中添加每个元素?

时间:2016-09-18 13:46:21

标签: python theano theano.scan

我想知道如何使用Theano在两个向量中添加每个元素?

假设我们有两个向量 vector_1 vecotr_2 ,我们想构建一个矩阵 A ,其中

  

A [i] [j] = vector_1 [i] + vecotr_2 [j]

我知道在numpy中我们可以使用列表理解。但我想使用Theano以更少的时间获得结果。似乎Theano.scan()可以完成这项工作,但我真的不知道如何处理它。

1 个答案:

答案 0 :(得分:0)

您可以使用广播。这是NumPy中的一个例子,你可以在Theano中做同样的事情:

>>> import numpy as np
>>> x1 = np.array([1,1,9]).reshape((3,1))
>>> x2 = np.array([0,3,4]).reshape((1,3))
>>> np.add(x1, x2)
array([[ 1,  4,  5],
       [ 1,  4,  5],
       [ 9, 12, 13]])