theano 共享变量在扫描函数中的形状错误

Question

我有一个形状为 (1, 500) 的 theano 共享变量，但是当传递给扫描函数时，形状变成了 (1, 1, 500)。示例代码 sn-p 如下。

y_t1 = theano.shared(name='y_t1', value=np.zeros((1, 500), dtype=theano.config.floatX))

def forward(X, y_t1):
    return y_t1

(hyp), _ = theano.scan(fn=forward, sequences=X, outputs_info=[y_t1])

y_t1 以 (1, 500) 的大小创建，并在函数“forward”之外报告其形状为 (1, 500)，但在“forward”内部它具有形状 (1, 1, 500)。为什么会这样？

谢谢。

Answer 1

把它作为

(hyp), _ = theano.scan(fn=forward, sequences=X, outputs_info=y_t1)

那么它应该可以正常工作。 （我已经删除了 output_info 中 y_t1 周围的括号）

解释：Theano 将你在 = 之后传入的任何内容转换为张量。因此，如果您传入一个列表，它首先会转换为该形状的张量。因此，当您传入 [y_t1] 时，您基本上是在为您的输入添加一个额外的维度。