從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)