pytorch
文章平均质量分 77
tmk_01
这个作者很懒,什么都没留下…
展开
-
pytorch方法测试——卷积(一维)
import torchimport torch.nn as nnm = nn.Conv1d(2, 5, 2, stride=2)print(m)input = torch.randn(2,2, 5)print(input)output = m(input)print(m.weight)print(m.bias)print(output)print(output.size()...原创 2018-06-12 19:22:48 · 12947 阅读 · 0 评论 -
pytorch方法测试——卷积(二维)
测试代码:import torchimport torch.nn as nnm = nn.Conv2d(2, 2, 3, stride=2)input = torch.randn(1, 2, 5, 7)output = m(input)print("输入图片:")print(input)print("卷积的权重:")print(m.weight)print("卷积的偏重:")...原创 2018-06-12 21:40:48 · 2530 阅读 · 0 评论 -
pytorch方法测试——卷积(三维)
测试代码:import torchimport torch.nn as nnimport torch.autograd as autogradm = nn.Conv3d(2, 1, 3, stride=2)input = autograd.Variable(torch.randn(1, 2, 3, 5, 6))output = m(input)print("输入图片:")prin...原创 2018-06-12 22:19:56 · 7716 阅读 · 0 评论 -
pytorch方法测试——归一化(BatchNorm2d)
测试代码:import torchimport torch.nn as nnm = nn.BatchNorm2d(2,affine=True) #权重w和偏重将被使用input = torch.randn(1,2,3,4)output = m(input)print("输入图片:")print(input)print("归一化权重:")print(m.weight)print...原创 2018-06-13 15:28:00 · 35427 阅读 · 4 评论 -
pytorch方法测试——激活函数(ReLU)
测试代码:import torchimport torch.nn as nn#inplace为True,将会改变输入的数据 ,否则不会改变原输入,只会产生新的输出m = nn.ReLU(inplace=True)input = torch.randn(7)print("输入处理前图片:")print(input)output = m(input)print("ReLU输出...原创 2018-06-13 15:42:46 · 31978 阅读 · 0 评论 -
pytorch方法测试——损失函数(CrossEntropyLoss)
测试代码:import torchimport torch.nn as nnimport mathloss = nn.CrossEntropyLoss()input = torch.randn(1, 5, requires_grad=True)target = torch.empty(1, dtype=torch.long).random_(5)output = loss(input,...原创 2018-06-28 10:57:08 · 29326 阅读 · 6 评论 -
pytorch方法测试——损失函数(BCELoss)
测试代码:import torchimport torch.nn as nnimport mathm = nn.Sigmoid()loss = nn.BCELoss(size_average=False, reduce=False)input = torch.randn(3, requires_grad=True)target = torch.empty(3).random_(2)...原创 2018-06-28 16:01:06 · 43130 阅读 · 2 评论