执行theano.function（）语句时出错

Question

我试图运行在Theano中实现的here提供的编码器解码器模型。

code snippet类中的以下SessionEncoderDecoder发出错误。

self.train_fn = theano.function(
                inputs=[self.x_data, self.x_ranks, self.x_max_length, self.x_cost_mask],
                outputs=self.training_cost, updates=self.updates, name="train_fn")

错误

Segmentation fault (core dumped)

SessionEncoderDecoder类中的__init__函数声明了各自的变量。

def __init__(self, state):
    ...
    self.x_data = T.imatrix('x_data')
    self.x_ranks = T.imatrix('x_ranks')
    self.x_cost_mask = T.matrix('cost_mask')
    self.x_max_length = T.iscalar('x_max_length')
    ...
    self.training_cost = T.sum(-T.log2(target_probs) * training_x_cost_mask) + np.float32(self.lambda_rank) * self.rank_cost
    self.updates = self.compute_updates(self.training_cost / training_x.shape[1], self.params)
    ...

所以，我不知道为什么我会出现分段错误。任何解决错误的线索都将非常感激。