深度学习与PyTorch笔记7

Tensor维度变换 view/reshape lost dim information view和reshape用法完全相同,将数据合并,必须要有物理意义。适用于全连接层。缺点:不好还原数据。 squeeze/unsqueeze squeeze:挤压、减少维度 unsqueeze:展开、增加维度,取值范围为[-a.dim()-1,a.dim()+1) 正数:在前面插入。负数:在后面插入。 exp
相关文章
相关标签/搜索