【人工智能】王小草的博客

微博：https://weibo.com/wangxiaocaoai/profile?rightmod=1&wvr=6&mod=personinfo微信公众号：搜索"AI躁动街"本节要点：1 张量2 创建张量的方式3 张量的索引,切片,连接,换位等操作4 随机抽样的操作5 序列化（保存与加载）6 并行化1 张量Tensor1.1 判断是否为张量t

方法一（推荐）只保存和加载模型的参数# 保存模型参数def save_model(the_model, PATH): torch.save(the_model.state_dict(), PATH)# 加载模型参数def load_model(PATH): the_model = TheModelClass(*args, **kwargs) the_model....

微博：https://weibo.com/wangxiaocaoai/profile?rightmod=1&wvr=6&mod=personinfo微信公众号：搜索"AI躁动街"本节要点：1 逐点计算操作2 缩减操作3 比较操作4 其他操作5 线性代数操作1 逐点计算操作# 导入包import torch# 1

微博：https://weibo.com/wangxiaocaoai/profile?rightmod=1&wvr=6&mod=personinfo微信公众号：搜索"AI躁动街"本节要点：1 张量的类型2 张量的创建3 张量的索引与切块4 张量的操作4.1 操作名有无下划线的区别4.2 torch.Tensor所有操作列表4.3 torch.Tensor操作举例...

一个torch.Storage是一个单一数据类型的连续一维数组。每个torch.Tensor都有一个对应的、相同数据类型的存储：class torch.FloatStorage1 torch.Storage类型import torchtorch.FloatStorage([1,2,3,4,5])torch.ByteStorage([1,2,3,4,5])torch.ShortStor...

本节要点：1 卷积层2 池化层3 非线性激活层4 正则层5 循环层6 线性层7 Dropout层8 Sparse层9 Veision层10 Multi-GPU层1 卷积层1.1 一维卷积层类名：class torch.nn.Conv1d(in_channels, out_channels, kernel_size, str...

1 class torch.nn.Moduletorch.nn.Module是所有网络的基类。我们创建的任何模型都应该继承这个类。Modules也可以包含其它Modules,允许使用树结构嵌入他们。你可以将子模块赋值给模型属性。1.1 构建模型现在，让我们来继承Module来建立一个超简单的模型吧。import torch.nn as nnimport torch.nn.functio...