python - 如何使用 Theano 形成复合函数？

Question

我想用 Theano 计算复合函数 f(x, g(x))。不幸的是，当我尝试编写函数组合代码时，Python 提示出现类型错误。例如，考虑以下简单脚本:

import theano
import theano.tensor as T

x = T.dscalar('x')

def g():
    y1 = T.sqr(x)
    return theano.function([x], y1)

def composition():
    input = g()
    yComp = x * input
    return  theano.function([x], yComp)

def f():
    y1 = T.sqr(x)
    yMult = x * y1
    return theano.function([x], yMult)

当编写 funComp = Composition() 时，Python返回一个TypeError:

TypeError: unsupported operand type(s) for *: 'TensorVariable' and 'Function' 

但是，我可以编译并计算函数 fun = f() 。有没有办法成功建立函数组合？我很感激任何帮助!

Answer 1

对于这种情况，您实际上不需要多个功能。这个效果很好。

import theano
import theano.tensor as T

x = T.dscalar('x')


def g():
    y1 = T.sqr(x)
    return y1

def composition():
    input = g()
    yComp = x * input
    return  theano.function([x], yComp)

tfunc = composition()
print tfunc(4)