pytorch的python API
文章平均质量分 58
机器视觉全栈er
**专注机器视觉全栈**----
www.去cvtutorials.com
编程语言:python,cplusplus,js;
软:vue,flask/django, opencv, pytorch/tf2, qt/pyqt, deepstream;
硬:工业相机,usb相机,英特尔realsense,微软k4a, jetson系列;
次要关注:
语音,nlp;
展开
-
pytorch的python API略读--Sequential/ModuleList/ModuleDict
作者:机器视觉全栈er网站:cvtutorials.comtorch.nn.Sequential:序列容器,顾名思义,就是将构造函数中的子模块会按照你定义的序列顺序被添加到模块中。这里有个注意的小点,要注意区分Sequential和torch.nn.ModuleList,后者就是一个简单的列表,里面的元素是模块,但是模块之间是孤立的,前者则是互相连通的模块。还是以上面的网络模块为例,使用Sequential可以实现快速搭建:import torch.nn as nnimport torchc.原创 2022-04-30 16:46:12 · 1733 阅读 · 0 评论 -
pytorch的python API略读--torch.nn.Module
作者:机器视觉全栈er网站:www.cvtutorials.com2.3 torch.nn本节我们看下torch.nn,这个模块是基础模块。2.3.1 容器torch.nn.Module:这个类是所有网络模块的基类,甚至你写普通的数学函数的时候,也可以用这个作为基类。内置的模块和你自定义的模块都可以相互嵌套,构成一个树状的模块结构。下面我们来看下这个Module类常用案例:import torchimport torch.nn as nnimport torch.nn.functional原创 2022-04-24 20:18:19 · 1217 阅读 · 0 评论 -
pytorch的python API略读--序列化和反序列化
作者:机器视觉全栈er网站:cvtutorials.com2.2 序列化和反序列化序列化是将某个对象的信息转化成可以存储或者传输的信息的过程,反序列化是和序列化相反的过程。torch.save: 序列化的用法如下:torch.save(obj, f, pickle_module=pickle, pickle_protocol=DEFAULT_PROTOCOL, _use_new_zipfile_serialization=True)举个简单的例子:>>> import .原创 2022-03-21 20:23:32 · 1881 阅读 · 0 评论 -
pytorch的python API略读--tensor(六)
作者:机器视觉全栈er网站:cvtutorials.com2.1.5 变换变换的概念更加宽泛一点,我们这里指索引、分和合以外的tensor操作。torch.gather:沿着某个维度对输入tensor进行变换,用法如下:torch.gather(input, dim, index, *, sparse_grad=False, out=None) 下面举个例子说明下:>>> cvtutorials = torch.randn(2, 2)>>> cvtu.原创 2022-03-06 17:33:08 · 948 阅读 · 0 评论 -
pytorch的python API略读--tensor(五)
pytorch中tensor的“合”的常用函数原创 2022-03-04 21:13:16 · 489 阅读 · 0 评论 -
pytorch的python API略读--tensor(四)
作者:机器视觉全栈er来源:cvtutorials.com2.1.3 分分就是将tensor拆分或者只看tensor的某个部分。torch.chunk:将一个tensor分解成多个块。用法如下:torch.chunk(input, chunks, dim=0)如果chunk不能被输入的tensor的dim方向上的整除的话,最后一个块和其他块的大小不一样,举个例子来说明下:>>> cvtutorials = torch.randn(7, 7)>>> c.原创 2022-03-01 21:44:02 · 461 阅读 · 0 评论 -
pytorch的python API略读--tensor(三)
作者:机器视觉全栈er来源:cvtutorials.com2.1.2 索引筛选出符合某种条件的subtensor。torch.where: 根据布尔变量的值选择tensor中的元素,用法如下:torch.where(condition, x, y)下面举个简单的例子:>>> import torch>>> cvtutorials = torch.randn(3, 4)>>> threshold = torch.zeros(3, 4.原创 2022-02-24 22:21:01 · 364 阅读 · 0 评论 -
pytorch的python API略读--tensor(二)
作者:机器视觉全栈er 来源:cvtutorials.com2.1.1 创建tensor在编写程序的时候,我们往往需要一些数据,这些数据常有两种来源:手动输入的一些数据,或者从外部接收后转换得到的数据。手动输入的数据设计如下函数:tensor, zeros, zeros_like, ones, ones_like,arange,range,linspace,logspace,eye,empty,empty_like,empty_strided,full,full_like。tensor:这是最基本.原创 2022-02-18 21:38:32 · 835 阅读 · 0 评论 -
pytorch的python API略读--tensor(一)
从pytorch的python API中选取常用数据类型判断及属性设置的函数进行解读。原创 2022-02-17 20:27:35 · 678 阅读 · 0 评论