pytorch
qq_35515203
这个作者很懒,什么都没留下…
展开
-
pytorch学习记录2--tensor维度变化
view / reshape #改变维度为指定维度 a=torch.rand(2,3,28,28) a.view(2,-1) .shape a.reshape(2,-1).shape 输出结果为: torch.Size([2, 2352]) torch.Size([2, 2352]) 2.squeeze / unsqueeze #压缩或扩展维度 squeeze...原创 2019-10-25 16:35:33 · 416 阅读 · 0 评论 -
pytorch学习记录1-------tensor基本操作
最近对pytorch进行了初步学习,本博客用于整理巩固学习成果,在此分享,希望与众佬共同进步。 pytorch中对tensor的基本操作主要分为,生成tensor、初始化tensor、从np导入tensor以及索引tensor。 1.生成tensor tensor作为众多深度学习框架的基本数据类型,也是pytorch的基本数据结构。在pytorch中生成tensor可以直接使用tor...原创 2019-10-24 20:53:02 · 257 阅读 · 0 评论