使用Theano.scan在条件上迭代ndarray

时间:2016-12-20 22:22:50

标签: python theano theano.scan

请原谅我一个基本问题(我是Theano的新手)!

我想只得到满足条件的那些位置的2个矩阵的差异。所以,假设我们有2个矩阵A和B,这个(python等价代码)是我想要计算的:

sum = 0
n,m = A.shape
for i in xrange(n):
    for j in xrange(m):
        if(A[i][j] != 3.5):        #some random condition!
            sum += A[i][j] - B[i][j]

我想要一个Theano等效代码来计算总和。 我知道有 Theano.scan 可用于扫描 ndarray ,但我无法获得任何具有if条件的示例。 提前谢谢你:)

0 个答案:

没有答案