我想在矩阵 X 上定义一些函数。例如 mean(pow(X - X0, 2)) ,其中 X0 是另一个矩阵( X0 是固定/常量)。为了更具体,让我们假设 XX0 都是 10 x 10 矩阵。运算的结果是一个实数。

现在我有一个大矩阵(比如说 500 x 500 )。我想将上面定义的操作应用于“大”矩阵的所有 10 x 10 子矩阵。换句话说,我想在“大”矩阵上滑动 10 x 10 窗口。对于窗口的每个位置,我应该得到一个实数。所以,作为最终结果,我需要得到一个实值矩阵(或二维张量)(它的形状应该是 491 x 491 )。

我想要的是接近卷积层但不完全相同,因为我想使用均方偏差而不是由神经元表示的线性函数。

最佳答案

这只是一个 Numpy 解决方案,希望它足够了。
我假设您的函数由对矩阵元素的运算和均值(即缩放和)组成。因此,查看 Y 就足够了

Y = np.power(X-X0, 2)

所以我们只需要处理确定一个加窗均值。请注意,对于一维情况,可以确定具有适当向量的矩阵乘积以计算平均值,例如
h = np.array([0, 1, 1, 0])  # same dimension as y
m1 = np.dot(h, y) / 2
m2 = (y[1] + y[2]) / 2
print(m1 == m2)  # True

2D 情况类似,但有两个矩阵乘法,一个用于行,一个用于列。例如。
m_2 = np.dot(np.dot(h, Y), h) / 2**2

要构建滑动窗口,我们需要构建一个移动窗口矩阵,例如
H = [[1, 1, 1, 0, 0, ..., 0],
     [0, 1, 1, 1, 0, ..., 0],
            .
            .
            .
     [0, ..., 0, 0, 1, 1, 1]]

计算所有的总和
S = np.dot(np.dot(H, Y), H.T)

带有 (n, n) 窗口的 (m, m) 矩阵的完整示例是
import numpy as np

n, m = 500, 10
X0 = np.ones((n, n))
X = np.random.rand(n, n)
Y = np.power(X-X0, 2)

h = np.concatenate((np.ones(m), np.zeros(n-m)))  # window at position 0
H = np.vstack((np.roll(h, k) for k in range(n+1-m)))  # slide the window
M = np.dot(np.dot(H,Y), H.T) / m**2  # calculate the mean
print(M.shape)  # (491, 491)

构建 H 的另一种但可能效率稍低的方法是
H = np.sum(np.diag(np.ones(n-k), k)[:-m+1, :] for k in range(m))

更新

使用该方法也可以计算均方偏差。为此,我们将向量恒等式 |x-x0|^2 = (x-x0).T (x-x0) = x.T x - 2 x0.T x + x0.T x0(空格表示标量或矩阵乘法,.T 表示转置向量)推广到矩阵情况:

我们假设 W 是一个包含块 (m,n) 单位矩阵的 (m.m) 矩阵,它能够通过 (k0,k1) 提取 (m,m) -th Y = W Z W.T 子矩阵,其中 Z 是包含数据的 (n,n) 矩阵。计算差异
 D = Y - X0 = Y = W Z W.T - X0

很简单,其中 X0D 是一个 (m,m) 矩阵。元素平方和的平方根称为 Frobenius norm 。基于这些 identities ,我们可以将平方和写为
s = sum_{i,j} D_{i,j}^2 = trace(D.T D) = trace((W Z W.T - X0).T (H Z H.T - X0))
  = trace(W Z.T W.T W Z W.T) - 2 trace(X0.T W Z W.T) + trace(X0.T X0)
  =: Y0 + Y1 + Y2
Y0 项可以解释为上述方法中的 H Z H.TY1 项可以解释为 Z 的加权平均值,Y2 是一个常数,只需要确定一次。
因此,一个可能的实现是:
import numpy as np

n, m = 500, 10
x0 = np.ones(m)
Z = np.random.rand(n, n)

Y0 = Z**2
h0 = np.concatenate((np.ones(m), np.zeros(n-m)))
H0 = np.vstack((np.roll(h0, k) for k in range(n+1-m)))
M0 = np.dot(np.dot(H0, Y0), H0.T)

h1 = np.concatenate((-2*x0, np.zeros(n-m)))
H1 = np.vstack((np.roll(h1, k) for k in range(n+1-m)))
M1 = np.dot(np.dot(H1, Z), H0.T)

Y2 = np.dot(x0, x0)
M = (M0 + M1) / m**2 + Y2

关于python - 如何使用 Theano 将 "arbitrary"操作放入滑动窗口?,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/41341973/

10-12 18:24