import numpy as np
a = np.array([[[0.1, 0.2, 0.3], [0.2, 0.3, 0.4]], [[0.4, 0.4, 0.4], [0.7, 
    0.6, 0.8]]], np.float32)
b = np.average(a, axis=2)
b
c = np.dstack((b, b, b))
c
средний массив numpy, но сохраняет форму