
pytorch
文章平均质量分 77
啾比特
这个作者很懒,什么都没留下…
展开
-
【Pytorch入门】合并与分割
Pytorch学习笔记——合并与分割 torch.cat torch.cat(tensors,dim=0,out=None) 作用:torch.cat是将两个tensor拼接在一起 torch.cat主要有两种用法,一种是按维数0拼接(按行拼接),另一种是按维数1拼接(按列拼接) 按维数0拼接(按行拼接) example 我们首先先创建两个二维张量x和y,分别是1行3列和2行3列 import torch x=torch.zeros(1,3)#1*3的tensor y=torch.zeros(2,3)#2*原创 2021-01-28 19:17:58 · 706 阅读 · 2 评论 -
【Anaconda】在Jupyter notebook中切换虚拟环境
切换环境 默认已经从创建好虚拟环境 1.打开Prompt,切换到你的虚拟环境 conda activate 你的虚拟环境的名字 2.安装ipykernel conda install ipykernel 3、将你的虚拟环境加入到ipykernel python -m ipykernel install --name 你的虚拟环境的名字 此时再打开Jupyter notebook就会出现你的虚拟环境了 遇到的问题 再次通过命令行打开jupyter notebook,报错 ...原创 2021-01-28 17:32:18 · 698 阅读 · 0 评论 -
【Pytorch入门】Tensor维度变换
Pytorch学习笔记——Tensor维度变换 view()/reshape() torch.view(arg1,arg2....) 作用:类似于numpy中的resize()的功能,重构tensor的维度,返回一个有相同数据但不同大小的tensor 简单来讲,就是把原先tensor中的数据按照行优先的顺序排成一个一维数组,然后根据参数组合成其他维度的tensor example: 首先我们构造两个张量a([1,2,3,4,5,6]),b([[1,2,3],[4,5,6]]),对其执行view操作后,输出结原创 2021-01-25 18:37:25 · 2892 阅读 · 0 评论