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

Question

让某些4D数组[x，y，z，k]包含k个3D图像[x，y，z]。 有什么方法可以计算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)