flatten(dim)是对多维数据的降维函数,默认缺省参数为0。 flatten(dim)表示,从第dim个维度开始展开,将后面的维度转化为一维。只保留dim之前的维度,其他维度的数据都展开到dim这一维。