从4D阵列计算3D像素方差

Question

假设存在由k个3D图像[x，y，z]组成的4D数组[x，y，z，k]。 有什么方法可以计算4D阵列中3D中每个像素的方差？

例如，我有一个10x10x10x5的数组，想返回一个10x10x10的方差数组； 计算沿k的每个像素（或像素）的方差

如果这没有道理，请告诉我，我会尽力解释。

目前，我的代码是：

tensors = []
while error > threshold:
    for _ in range(5): #arbitrary
        new_tensor = foo(bar) #always returns array of same size
        tensors.append(new_tensor)

tensors = np.stack(tensors, axis = 3)
#tensors.shape

我想计算张量的方差数组

Answer 1

如果您使用的是numpy，则有一种简单的方法：

variance = tensors.var(axis=3)