Xiao_Bai_Ke的博客

''' 使用pytorch的自动求导机制 ''' import torch dtype=torch.float device=torch.device("cpu") # N是批量大小; D_in是输入维度; # H是隐藏的维度; D_out是输出维度。 N,D_in,H,D_out=64,1000,100,10 #随机构建输入输出 x=torch.randn(N,D_in,device=...

import torch dtype=torch.float device=torch.device("cpu") # N是批量大小; D_in是输入维度; # H是隐藏的维度; D_out是输出维度。 N,D_in,H,D_out=64,1000,100,10 #创建随机输入和输出数据 x=torch.randn(N,D_in,device=device,dtype=dtype) y=torc...

import torch import torchvision import torch.nn as nn import torch.nn.functional as F import torch.optim as optim import torchvision.transforms as transforms #归一化 transform = transforms.Compose( ...