Могу ли я сохранить промежуточные результаты?

Question

Я хочу сохранить промежуточные результаты, чтобы избежать нескольких вычислений для одной вещи. То, что я ищу что-то вроде этого:

h1_activ = sigmoid(self.bias_visiblie + T.dot(D, self.W)) 
    h1_sample = h1_activ > rnds.uniform((n_samples, self.n_hidden)) 

    f_h1_sample = theano.function(
     inputs=[D], 
     outputs=h1_sample, 
     # I'd like to take the result from 'h1_sample' and store it into 'H1_sample' 
     updates=[(self.H1_sample, ???)] 
    ) 

Код выше не работает, конечно, но есть способ, чтобы сделать что-то вроде этого? Сохранение промежуточного значения в общую переменную?

Answer 1

Вы можете написать окончательные результаты, которые используют те же промежуточные результаты, в том же theano.function.

Например:

h1_activ = sigmoid(self.bias_visiblie + T.dot(D, self.W)) 
h1_sample = h1_activ > rnds.uniform((n_samples, self.n_hidden)) 
# h2_sample use the intermediate result h1_sample. 
h2_sample = h1_sample * 2 

f_h1_sample = theano.function(
    inputs=[D], 
    outputs=[h1_sample, h2_sample], 
) 

h2_smaple является конечным результатом, который использует h1_sample.

Также вы можете сохранить промежуточные результаты и использовать их в качестве входов в другом theano.function.

Разное theano.function s соответствуют разным графикам вычислений. Я думаю, что никакие вычисления не могут быть разделены между различными графиками вычислений.