对两个flatten函数的探究
1 问题
对torch.nn.flatten()和torch.flatten()两个函数的理解。
2 方法
对于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()函数官方文档： |
|---|
3 结语
通过对照实验，对两个函数的参数进行比照分析，得出结论。
本文参与 腾讯云自媒体分享计划，分享自微信公众号。
原始发表：2023-05-22，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读