windfly17的博客

简单训练一个分类器 import torch import torch.nn as nn import torch.nn.functional as F import torch.optim as optim import torchvision import torchvision.transforms as transforms import matplotlib.pyplot as plt import numpy as np transform = 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,