使用Theano扫描功能进行简单累加时遇到的问题

Question

我正在尝试累积矩阵值，并与步骤相乘： res = sum_i（i * a）。我的代码如下所示：

import numpy as np 
from theano import function, scan 
import theano.tensor as T 

x = T.lmatrix() 
results, updates = scan(
    lambda res, step, x: res + step * x, 
    non_sequences=x, 
    sequences=T.arange(2), 
    outputs_info=T.zeros_like(x)) 

f = function([x], results) 
a = np.array([[0, 0], [2, 2]], 'int64') 
print(f(a)) 

此输出：

[[[0 0] 
    [0 0]] 

[[1 1] 
[1 1]]] 

尽管我希望这样的：

[[[0 0] 
    [0 0]] 

[[0 0] 
[2 2]]] 

Answer 1

输出是（或许令人吃惊？）是正确的。你得到这个输出的原因如下：

在第一次迭代，

res = 0 
step = [[0, 0], [0, 0]] 
x = [[0, 0], [2, 2]] 

等

res + step * x = 0 + [[0, 0], [0, 0]] * [[0, 0], [2, 2]] 
       = 0 + [[0, 0], [0, 0]] 
       = [[0, 0], [0, 0]] 

在第二次迭代，

res = 1 
step = [[0, 0], [0, 0]] 
x = [[0, 0], [2, 2]] 

等

res + step * x = 1 + [[0, 0], [0, 0]] * [[0, 0], [2, 2]] 
       = 1 + [[0, 0], [0, 0]] 
       = [[1, 1], [1, 1]] 

请注意，1广播的形状与step和x的元素相乘得到的矩阵相同。