上传者: 38685455
|
上传时间: 2023-03-26 10:11:02
|
文件大小: 64KB
|
文件类型: PDF
代码已经调通,跑出来的效果如下:
# coding=gbk
import torch
import matplotlib.pyplot as plt
from torch.autograd import Variable
import torch.nn.functional as F
'''
Pytorch是一个拥有强力GPU加速的张量和动态构建网络的库,其主要构建是张量,所以可以把PyTorch当做Numpy
来用,Pytorch的很多操作好比Numpy都是类似的,但是其能够在GPU上运行,所以有着比Numpy快很多倍的速度。
训练完了,发现隐层越大,拟合的速度越是快,拟合的效果越