Pytorch搭建入门级网络

作为一个只学习了深度学习的理论基础,实操经验为0的小白,想通过输出的方式巩固一下搭网络的技能,并给同样是新手的同学一个比较基础的入门实例,这一篇是pytorch框架下搭网络的示例,包括使用pytorch自己的一套工具搭建,和只使用pytorch的autograd机制来搭建网络。

在阅读实例之前,推荐看一下pytorch的autograd的官方文档,或者中文翻译,有助于理解。

注:以下代码有部分参考莫烦的pytorch教程中的分类网络,有兴趣的同学也可以看看他的教程

话不多说,开始上代码。

一、完全使用pytorch框架搭建网络

import numpy as np
import matplotlib.pyplot as plt

//生成200个数据点
n_data = torch.ones(100, 2)
x0 = torch.normal(2*n_data, 1)      # class0 x data (tensor), shape=(100, 2)
y0 = torch.zeros(100)               # class0 y data (tensor), shape=(100, 1)
x1 = torch.normal(-2*n_data, 1)     # class1 x data (tensor), shape=(100, 2)
y1 = torch.ones(100

你可能感兴趣的:(入门教程,python,神经网络,深度学习)