gpt4 book ai didi

theano - 如何在不重建图形的情况下重用具有不同共享变量的 Theano 函数?

转载 作者:行者123 更新时间:2023-12-04 15:23:47 24 4
gpt4 key购买 nike

我有一个被多次调用的 Theano 函数,每次都使用不同的共享变量。按照现在的实现方式,Theano 函数在每次运行时都会重新定义。我假设,这会使整个程序变慢,因为每次定义 Theano 函数时,都会重建图形。

def sumprod_shared(T_shared_array1, T_shared_array2):
f = theano.function([], (T_shared_array1 * T_shared_array2).sum(axis=0))
return f()

for factor in range(10):
m1 = theano.shared(factor * array([[1, 2, 4], [5, 6, 7]]))
m2 = theano.shared(factor * array([[1, 2, 4], [5, 6, 7]]))
print sumprod_shared(m1, m2)

对于非共享(普通)变量,我可以定义一次函数,然后使用不同的变量调用它而无需重新定义。

def sumprod_init():
T_matrix1 = T.lmatrix('T_matrix1')
T_matrix2 = T.lmatrix('T_matrix2')
return theano.function([T_matrix1, T_matrix2], (T_matrix1 * T_matrix2).sum(axis=0))

sumprod = sumprod_init()
for factor in range(10):
np_array1 = factor * array([[1, 2, 4], [5, 6, 7]])
np_array2 = factor * array([[1, 2, 4], [5, 6, 7]])
print sumprod(np_array1, np_array2)

这也适用于共享变量吗?

最佳答案

你可以在 theano.function 中使用 givens 关键字。基本上,您执行以下操作。

m1 = theano.shared(name='m1', value = np.zeros((3,2)) )
m2 = theano.shared(name='m2', value = np.zeros((3,2)) )

x1 = theano.tensor.dmatrix('x1')
x2 = theano.tensor.dmatrix('x2')

y = (x1*x2).sum(axis=0)
f = theano.function([],y,givens=[(x1,m1),(x2,m2)],on_unused_input='ignore')

然后循环遍历值,您只需将共享变量的值设置为您想要的值。顺便说一下,您必须将 on_unused_input 设置为 'ignore' 以使用在 theano 中没有参数的函数。像这样:

array1 = array([[1,2,3],[4,5,6]])
array2 = array([[2,4,6],[8,10,12]])

for i in range(10):
m1.set_value(i*array1)
m2.set_value(i*array2)
print f()

它应该有效,至少我是这样解决的。

关于theano - 如何在不重建图形的情况下重用具有不同共享变量的 Theano 函数?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/17039908/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com