def METHOD_NAME(x: NNCFTensor, axis: int) -> NNCFTensor:
    if len(x.shape) < 3:
        return ONNXNNCFTensor(np.mean(x.tensor, axis=0))
    x = np.moveaxis(x.tensor, axis, 1)
    t = x.reshape(x.shape[0], x.shape[1], -1)
    return ONNXNNCFTensor(np.mean(t, axis=(0, 2)))