Pytorch
yuki___
这个作者很懒,什么都没留下…
展开
-
Pytorch入门
Pytorch学习-1基础tensor基本语法(与NumPy相似)Tensor的运算一个简易神经网络导入包初始化权重定义训练次数和学习效率梯度下降优化神经网络的参数自动梯度torch.autograd和Variable导入包初始化权重定义训练次数和学习效率新的模型训练和参数优化自定义传播函数转载自 基础 tensor基本语法(与NumPy相似) torch.FloatTensor,传递给torc...转载 2019-07-20 18:14:00 · 899 阅读 · 0 评论 -
Pytorch的nn.module
包引入 import torch import numpy as np import torch.nn as nn import torch.nn.functional as F import torch.optim as optim from torch.autograd import Variable 定义类 class net(nn.Module): def __init__(self)...转载 2019-07-22 14:30:29 · 210 阅读 · 0 评论 -
Pythorch的nn.functional和Parameter
Convolution函数 2D卷积 torch.nn.functional.conv2d(input, weight, bias=None, stride=1, padding=0, dilation=1, groups=1) 参数: input – 输入张量 (minibatch x in_channels x iH x iW) weight – 过滤器张量 (out_channels, ...转载 2019-07-22 17:09:18 · 151 阅读 · 0 评论