python - 如何从 theano 函数内部打印值？

Question

我最近从 Matlab/C++ 转移到 theano 并具有以下功能

train_model = theano.function([x_in, y_index],
                          classifier.cost,
                          updates=updates,
                          givens={
                              x: x_in,
                              y: y_in[y_index]})

我想在每次迭代中打印两层网络之间的值(用于调试、更好地控制功能等)我已经尝试编辑设置分类器的功能，以便它打印(使用 print() 或 theano.printing.Print/theano.pp())，我得到的只是在设置模型时的单个打印。

Answer 1

在您的示例中，classifier.cost 是一个表达式，可能由基于同一输入的其他几个表达式组成。您可以像使用 classifier.cost 一样将任何这些中间表达式转换为函数，例如

f_first_layer = theano.function([x], first_layer)

然后您可以调用并打印此函数的输出，例如在每次调用 train_model 之后。如果你在 before train_model 中调用它时使用与调用 train_model 时相同的参数，那么你将得到层的精确输出因为它们将由 train_model 进行评估(在 train_model 之后调用它会因更新而有所不同)。