Theano:ifelse为三个案例

时间:2016-05-05 06:24:25

标签: if-statement neural-network theano

我试图实现this paper中描述的神经网络模型。然而,损失函数由4部分if / else块组成,类似于这种结构:

if correct: loss = 0
elif <condition1>: loss = 0.5
elif <condition2>: loss = 0.2
else: loss = 0.4

我知道theano.ifelse.ifelse操作。但是,为了实现此结构,将有四个嵌套ifelse个案例。有没有更简单的方法来实现这四种情况?

(为了记录,我实际上在theano中实现了嵌套ifelse,但我遇到了与this google groups post相同的错误。)

0 个答案:

没有答案