flatten函数
将维度展开为一维的函数。
python里的flatten(dim)表示,从第dim个维度开始展开,将后面的维度转化为一维。
也就是说,只保留dim之前的维度,其他维度的数据全都挤在dim这一维。
X.flatten(2)
此处的flatten(2)表示将第2个维度及以后维度展开到第2维上。
flatten函数
将维度展开为一维的函数。
python里的flatten(dim)表示,从第dim个维度开始展开,将后面的维度转化为一维。
也就是说,只保留dim之前的维度,其他维度的数据全都挤在dim这一维。
X.flatten(2)
此处的flatten(2)表示将第2个维度及以后维度展开到第2维上。