gee 的flatten

`flatten()` 是一个在 numpy 或 TensorFlow 等库中经常使用的函数，用于将多维数组展平为一维数组。在 Gee 中，`flatten()` 是一个用于将图片数据转化为一维向量的函数。具体来说，如果一个图片的大小为 (height, width, channel)，那么 `flatten()` 将把它转化为一个大小为 (height * width * channel,) 的一维向量。

这个函数在 Gee 的实现中比较简单，可以用以下代码实现：

def flatten(x):
    """
    将输入的多维数组 x 展平为一维向量
    """
    return x.reshape(x.shape[0], -1)

其中，`reshape()` 函数是 numpy 中用于改变数组形状的函数，第一个参数表示新数组的形状，第二个参数表示自动计算该维度的大小。在这里，我们将多维数组的第一个维度保留不变，将剩下的维度展平为一维。