摘要: # -*- coding: utf-8 -*- import torch from torch.autograd import Variable dtype = torch.FloatTensor # dtype = torch.cuda.FloatTensor # Uncomment this to run on GPU # N is batch size; D_in is input d... 阅读全文
posted @ 2017-06-11 15:14 hopskin1 阅读(852) 评论(0) 推荐(0) 编辑
摘要: [1]中的程序可以改成如下对应的Tensor形式: 有两个函数需要说明 h.clamp(min=0) clamp表示夹紧,夹住的意思,torch.clamp(input,min,max,out=None)-> Tensor 将input中的元素限制在[min,max]范围内并返回一个Tensor 用 阅读全文
posted @ 2017-06-11 11:47 hopskin1 阅读(8535) 评论(0) 推荐(0) 编辑
摘要: torch的第一个例子程序,是用numpy函数实现神经网络。cs231n的课程中有大量这样的作业。 这是一个三层的神经网络,包括一个输入层,一个中间隐藏层和一个输出层,神经元数分别为Din, H和Dout 前向传播时,H=XW1, H=max(H,0), Y=HW2 反向传播时,按如下公式计算梯度: 阅读全文
posted @ 2017-06-11 10:37 hopskin1 阅读(2592) 评论(0) 推荐(0) 编辑