
pytorch
文章平均质量分 56
1
不再彷惶
优快云已不再更新,所有内容迁移到了个人博客网站:bilicili.com
展开
-
pytorch基础
持续补充import torch1.随机相关a = torch.randn(2,3) 正态分布b = torch.rand(2,3) 0~1范围内随机c = torch.rand_like(a) 模仿a的形状生成随机矩阵d = torch.randint(1,10,[2,3,4]) 在1~10内生成形状为(2,3,4)的矩阵a = a.cuda() 将数据加载到gpu内2.查看数据形状a.type() 输出数据类型a.size() == a.shape原创 2020-10-21 16:51:38 · 2469 阅读 · 0 评论 -
MNIST手写数字识别总结(pytorch)
此博客并不是教程,只是一个练习总结 代码汇总放在文末 1.首先导入所需要的库import torchfrom torch import nnfrom torch.nn import functional as Ffrom torch import optimimport torchvisionfrom matplotlib import pyplot as pltimport pandas as pdimport numpy as npfrom Util import plot.原创 2020-10-21 16:06:35 · 2562 阅读 · 0 评论