pytorch学习笔记(1)
#pytorch 的nn库
import torch.nn as nn
N,D_in,H,D_out=64,1000,100,10
#随机创建一些训练数据
x=torch.randn(N,D_in)
y=torch.randn(N,D_out)
# 可以定义一个model
model=torch.nn.Sequential(
#一个线性层,一个relu函数,一个线性层
torch.nn.Linea...
原创
2020-03-10 17:53:48 ·
408 阅读 ·
0 评论