![](https://img-blog.csdnimg.cn/20190918140053667.png?x-oss-process=image/resize,m_fixed,h_224,w_224)
修仙之路:pytorch篇
好好学习,天天向上
宁静致远*
这个作者很懒,什么都没留下…
展开
-
Pytorch:torch.cat()
1torch.cat()在给定维度上对输入的张量序列seq进行连接操作。torch.cat()可以看做 torch.split() 和 torch.chunk()的反操作。 cat() 函数可以通过下面例子更好的理解。torch.cat(seq, dim=0, out=None) → Tensor参数 描述 seq(Tensors的序列) 可以是相同类型的Tensor的任何python序列 dim(int,可选) 沿着此维连接张量序列 out(Tensor,可..原创 2020-06-05 23:04:51 · 1155 阅读 · 0 评论 -
Pytorch:torch.mul() 、torch.mm()、torch.bmm()、torch.matmul()
1torch.mul() 用标量值value乘以输入input的每个元素,并返回一个新的结果张量。 \( out=tensor ∗ value \)。如果输入是FloatTensor or DoubleTensor类型,则value必须为实数,否则须为整数。 torch.mul(input, value, out=None)参数 描述 input(Tensor) 输入张量 value (Number) 乘到每个元素的数 out (Tensor) ...原创 2020-06-05 22:28:32 · 2326 阅读 · 1 评论 -
Pytorch:torch.ge()、torch.gt()、torch.le()、torch.lt()
1torch.ge()逐元素比较input和other,即是否 \( input >= other \)。如果两个张量有相同的形状和元素值,则返回True,否则False。 第二个参数可以为一个数或与第一个参数相同形状和类型的张量。torch.ge(input, other, out=None) → Tensor参数 描述 返回值 input 待对比的张量 一个torch.ByteTensor张量,包含了每个位置的比较结果(是否 input ...原创 2020-06-05 22:11:19 · 13846 阅读 · 3 评论 -
Pytorch:模型的保存与加载 torch.save()、torch.load()、torch.nn.Module.load_state_dict()
Pytorch 保存和加载模型后缀:.pt 和.pth1 torch.save()[source]保存一个序列化(serialized)的目标到磁盘。函数使用了Python的pickle程序用于序列化。模型(models),张量(tensors)和文件夹(dictionaries)都是可以用这个函数保存的目标类型。torch.save(obj, f, pickle_module=<module '...'>, pickle_protocol=2)参数 描述 o..原创 2020-06-05 12:46:18 · 200782 阅读 · 30 评论 -
numpy和tensor互转
1 torch.from_nump()功能:完成数组numpy到tensor的转换。栗子:import torchimport numpy as npa = np.ones(5)b = torch.from_numpy(a)print(type(a))print(type(b))输出结果:<class 'numpy.ndarray'><class 'torch.Tensor'>注意:这个函数所产生的的Tensor和NumPy中的数原创 2020-05-31 19:02:55 · 1411 阅读 · 0 评论 -
Pytorch:如何定义自己创建的数据集
本文为个人知识学习的记录,未来可以复习回顾。在Pytorch中定义数据集主要涉及到两个主要的类: Datasets DataLoader 1 Datasets1.1 什么是Datasets?Datasets是我们用的数据集的库,pytorch自带多种数据集,如Cifar10、MNIST等1.2为什么要定义Datasets?Pytorch中有工具函数torch.utils.Data.DataLoader,通过这个函数我们在准备加载数据集使用mini-batch的.原创 2020-05-17 11:52:55 · 7224 阅读 · 0 评论 -
Pytorch:搭建CNN模型的基本架构
import torch.nn as nn# 搭建CNN模型class CNN(nn.Module): def __init__(self): super(CNN, self).__init__() # 1号网络 self.conv1 = nn.Sequential( # 卷积参数设置 nn.Conv2d( in_channels=1, # 输入数据的通道为1,.原创 2020-05-13 12:45:32 · 3041 阅读 · 1 评论 -
Pytorch:torchvision.transforms
torchvision.transforms是专门用来对数据进行各种的处理。包括如下操作:归一化 PIL.Image / numpy.ndarray 与Tensor的相互转化 对PIL.Image进行裁剪、缩放等操作通常,在使用torchvision.transforms,我们通常使用transforms.Compose将transforms组合在一起。1torchvision.transforms.ToTensor()对于一个图片img,调用ToTensor转化成张量的形式,发生的.原创 2020-05-14 20:37:09 · 886 阅读 · 2 评论 -
Pytorch:torchvision.datasets
中文文档:https://ptorch.com/docs/8/torch-torchvisiontorchvision包括了目前流行的数据集,模型结构和常用的图片转换工具,是PyTorch中专门用来处理图像的库。这个包中有四个大类:torchvision.datasets torchvision.models torchvision.transforms torchvision.u...原创 2020-05-07 17:36:47 · 1275 阅读 · 1 评论