【深度学习】torch.nn.Flatten()和torch.flatten()的区别

torch.nn.Flatten 和 torch.flatten两个函数的区别

在深度学习模型训练和测试之前，通常要对tensor数据进行预处理，在处理过程中，涉及到将高纬度数据降维的操作。

torch.nn.Flatten 和 torch.flatten 都被用来进行降维操作。

torch.nn.Flatten() 有如下性质：

默认参数为start_dim = 1 , end_dim = -1，即从第1维（第0维不变）开始到最后一维结束将每个batch拉伸成一维：


当仅设置一个参数时，该参数表示 start_dim 的值，即从该维度开始到最后一个维度结束，将每个batch拉伸成一维，其余维度不变：

当设置两个参数时，两个参数分别表示开始维度和结束维度：

torch.nn.Flatten()函数官方文档：

对于torch.flatten():

torch.flatten()函数默认start_dim = 0 , 其余与torch.nn.flatten()相同。

torch.flatten()函数官方文档：