用其他theano函数输出定义theano函数

Question

我是提亚诺的新手，有人能帮我定义这样的提亚诺函数吗？

基本上，我有一个网络模型，如下所示：

y_hat, cost, mu, output_hiddens, cells = nn_f(x, y, in_size, out_size, hidden_size, layer_models, 'MDN', training=False)

这里，输入x是一个张量：

x = tensor.tensor3('features', dtype=theano.config.floatX)

我想为以后使用定义两个theano函数：

f_x_hidden = theano.function([x], [output_hiddens])

f_hidden_mu = theano.function([output_hiddens], [mu], on_unused_input = 'warn')

第一个很好。对于第二个问题，问题是输入和输出都是原始函数的输出。它给了我错误：

theano.gof.fg.MissingInputError:图的输入，用于计算Elemwise{identity}(特性)，没有提供，也没有给出值。

我的理解是，[output_hiddens]和[mu]都与输入[x]有关，它们之间应该有某种关系。我尝试将另一个theano函数从[x]定义为[mu]，例如：

f_x_mu = theano.function([x], [mu]), 

然后

f_hidden_mu = theano.function(f_x_hidden, f_x_mu), 

但还是不起作用。有人能帮我吗？谢谢。

Answer 1

简单的答案是不可能的。在这里中

因为在Theano中，首先以象征的方式表达一切，然后编译这个表达式以获得函数，

不能将theano.function的输出用作另一个theano.function的输入/输出，因为它们已经是编译过的图形/函数。

您应该在f_x_hidden的示例代码中传递符号变量(例如x)来构建模型。