windfly17的博客

简单训练一个分类器import torchimport torch.nn as nnimport torch.nn.functional as Fimport torch.optim as optimimport torchvisionimport torchvision.transforms as transformsimport matplotlib.pyplot as pltimport numpy as nptransform = transforms.Compose([tran

1. 张量1.初始化torch.rand(col, row) # 使用(0~1)均匀分布随机初始化二维数组torch.ones(col, row) # 填充1初始化torch.zero(col, row) # 填充0初始化torch.eye(col, row) # 对角线为一，其它为0初始化2.基本类型tensor = torch.tensor([3.1433223]) torch.DoubleTensor(col, row) # 64位浮点torch.FloatTensor(col,