pytorch
ncbnwk
这个作者很懒,什么都没留下…
展开
-
pytorch基础学习1
1.从numpy导入import torchimport numpy as npa=np.array([2,2.3])b=torch.from_numpy(a)print(a)print(b)a=np.ones([2,3])b=torch.from_numpy(a)print(a)print(b)结果:[2. 2.3]tensor([2.0000, 2.3000],...原创 2019-07-30 13:00:38 · 92 阅读 · 0 评论 -
pytorch-7交叉商1
定义:def cross_entropy(input, target, weight=None, size_average=None, ignore_index=-100, reduce=None, reduction='mean'): # type: (Tensor, Tensor, Optional[Tensor], Optional[boo...原创 2019-08-01 21:16:28 · 225 阅读 · 0 评论 -
pytorch-6线性优化
1.可视化,meshgrid合并x,ydef fun(x): return (x[0]**2+x[1]-11)**2+(x[0]+x[1]**2-7)**2x=np.arange(-6,6,0.1)y=np.arange(-6,6,0.1)X,Y=np.meshgrid(x,y)print(X.shape)print(Y.shape)Z=fun([X,Y])fig=p...原创 2019-08-01 20:53:02 · 168 阅读 · 0 评论 -
pytorch5求导
import torchfrom torch.nn import functionalx=torch.ones(1,requires_grad=True)print(x)w=torch.full([1],6,requires_grad=True)mse=functional.mse_loss(x,x*w)print(mse)#(1-6)**2print(torch.autograd...原创 2019-08-01 00:32:03 · 102 阅读 · 0 评论 -
pytorch基础学习5
1.whereimport torchcond=torch.tensor([[0.6,0.4],[0.7,0.3]])a=torch.tensor([[0.,0.],[0.,0.]])b=torch.tensor([[1.,1.],[1.,1.]])c=torch.where(cond>0.5,a,b)print(a)print(b)print(c)结果:tensor...原创 2019-07-31 22:15:26 · 103 阅读 · 0 评论 -
pytourch基础学习4
1.matmul/@/mm(二维)矩阵相乘import torcha=torch.randn(4,2,3,8)b=torch.randn(4,2,8,8)print(torch.matmul(a,b).shape)结果:torch.Size([4, 2, 3, 8])2.pow/**import torcha=torch.full([2,2],9)print(a)b...原创 2019-07-31 21:21:00 · 302 阅读 · 0 评论 -
pytourch基础学习3
1.合并import torcha=torch.randn(5,2,3,8)b=torch.randn(4,2,3,8)print(torch.cat([a,b],dim=0).shape)结果:torch.Size([9, 2, 3, 8])2.stack增加维度import torcha=torch.randn(4,2,3,8)b=torch.randn(4,2,...原创 2019-07-31 20:31:07 · 216 阅读 · 0 评论 -
pytorch基础学习2
1.切片import torchb=torch.tensor([[[1, 2, 3,4]]])print(b)a=b[:,:,0:4:2].shapeprint(a)print(b[:,:,0:4:2])print(b)结果:tensor([[[1, 2, 3, 4]]])torch.Size([1, 1, 2])tensor([[[1, 3]]])tensor([[[...原创 2019-07-30 14:22:51 · 88 阅读 · 0 评论