搜索结果
查询Tags标签： torch，共有 290条记录-
关于数学运算
加、减、乘、除这里直接用 符号 + - * / 就可以运算，结果和用函数是一样的，不过都要满足能运算的条件，满足broadcasting， a = torch.rand(3,4) b = torch.rand(4) c1 = a + b c2 = torch.add(a,b) d = torch.all(torch.eq(a-b,torch.sub(a,b))) #判断两种方法是否殊途…
2022/1/31 23:17:33 人评论 次浏览 -
Proximal Policy Optimization(PPO)算法实现gym连续动作空间任务Pendulum-v0（pytorch）
目录 1.ppo算法概述 2.Pendulum-v0 3.代码实现1.ppo算法概述PG算法 视频参考李宏毅强化学习课程：李宏毅深度强化学习(国语)课程(2018)_哔哩哔哩_bilibili 上图表示actor与环境交互的一次经过，从开始的状态s1，actor输出a1到环境状态变为s2...直到…
2022/1/31 17:11:13 人评论 次浏览 -
torch.nn.modules类构建模型
一、torch.nn.modules网络模型构建于一系列：nn.Conv2d、AvgPool2d、ReLU6、BatchNorm2d、CrossEntropyLoss等，由此基本操作组成。先看操作类型，再看操作组成方式。 1、属于nn.modules的操作类型 ① class torch.nn.Conv2d(in_channels, out_channels, kernel_size, str…
2022/1/31 6:05:54 人评论 次浏览 -
Broadcasting、合并and拆分
broadcasting它的功能就是能够实现自动维度的扩展，使两个tensor的shape相同， 也就是说节省了 unsqueeze 和 expand 操作，高效并且不需要复制数据，节省内存关键思想 只有size一致才能进行对应位置元素的相加如下图，第一行，两个tensor的shape相等，可以直接相加 第…
2022/1/30 23:08:52 人评论 次浏览 -
维度的变换
view 和 reshape 操作二者的功能和用法是相同的，不过要保证 view 的size整体不变，即numel（）一致 view操作会把维度信息给丢失掉，要记牢数据的存储顺序 a = torch.rand(4,1,28,28) b = a.view(4,28*28) #丢失两个维度 print(b) print(b.shape) c = a.view(4*28,28…
2022/1/29 23:07:15 人评论 次浏览 -
windows安装cpu版本的torch-geometric教程（附对应版本torch）
前提：conda环境 1、下载到本地 1.1 下载torch torch下载网址 图中圈起来的部分是windows版本的torch，根据自己的python版本下载，例如cp38代表python3.8 1.2 下载torch-geometric依赖包 torch-geometric依赖包下载地址 选择torch-1.5.0+cpu 进入页面后，分别下载torch_…
2022/1/29 7:04:38 人评论 次浏览 -
神经网络基本骨架、结构的使用
目录神经骨架的使用 基本神经结构的使用神经骨架的使用首先我们进官网进行相关信息的查询：container（相当于一个骨架）之中有六个模块，最常用的是module模块，因为对所有神经网络提供了一个最基本的类。 根据module的介绍，我们知道自己创建的model也需要继承该…
2022/1/26 23:05:37 人评论 次浏览 -
[源码分析] Facebook如何训练超大模型--- (5)
[源码分析] Facebook如何训练超大模型--- (5) 目录[源码分析] Facebook如何训练超大模型--- (5)0x00 摘要0x01 背景0x02 思路2.1 学习建议2.2 具体思路2.3 最佳实践0x03 具体实现3.1 Wrapper3.2 如何使用3.2 _checkpointed_forward3.2.1 处理输入3.2.2 非张量输出3.2.2.1 …
2022/1/26 9:04:19 人评论 次浏览 -
PyTorch中卷积的简单操作
卷积 Conv1d Conv1d input：形状的输入张量 weight： 形状过滤器 bias：形状的可选偏置张量( out_channels ). 默认：None stride：卷积核的步长。可以是单个数字或元组(sH, sW)。默认值：1 padding：输入两侧的隐式填充。可以是字符串 {‘valid’, ‘same’}、单个数…
2022/1/26 6:06:56 人评论 次浏览 -
LIBTORCH C++
1:Ubuntu18.04 配置 LibTorch_不忘初心~-CSDN博客_ubuntu安装libtorch 2:You need C++14 to compile PyTorch Ubuntu下C++调用pytorch训练好模型--利用libtorch - 灰信网（软件开发博客聚合） 3:Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor)…
2022/1/25 20:04:26 人评论 次浏览 -
ubuntu20.04 使用libtorch
一、官网下载文件 https://pytorch.org/get-started/locally/ 选择下载Stable(1.10.1) C++ CPU(cxx 11 ABI) 解压.zip压缩包，将libtorch移动到存放第三方库的位置 二、使用 2.1 创建torch_ws文件夹，然后创建如下两文件 example-app.cpp #include "torch/torch.h&qu…
2022/1/25 7:07:40 人评论 次浏览 -
[源码分析] Facebook如何训练超大模型---(4)
[源码分析] Facebook如何训练超大模型 --- (4) 目录[源码分析] Facebook如何训练超大模型 --- (4)0x00 摘要0x01 背景知识1.1 单精度、双精度和半精度浮点格式的区别1.2 多精度和混合精度计算的区别1.3 混合精度1.4 训练过程0x02 PyTorch2.1 英伟达算力2.2 Torch.cuda.a…
2022/1/24 20:05:01 人评论 次浏览 -
python torch节约内存开销的运算方法
示例如下，首先创建两个两行一列，值全为1的张量a和b，然后将b加到a上，使用python自带的id函数来获取a和b的内存地址。下面依次演示不同方法获得结果的差异： 1. +运算 a = torch.ones(2,1) b = torch.ones(2,1) before_id = id(a) a = a + b after_id = id(a) befo…
2022/1/24 7:07:27 人评论 次浏览 -
解决no module named torch
Anaconda里的python和已安装的python冲突， 在应用里把python卸了，python的环境变量也删了如图，anaconda文件夹中已有python.exe，只有卸载我们安装的python，vscode才能调用此python
2022/1/24 6:06:09 人评论 次浏览 -
torch.gather与torch.scatter
前言 我觉得pytorch最难的两个api之一就是gather和scatter了，但是在项目中又经常出现，OMG… 还是想办法弄懂吧torch.gather input = torch.randn((2, 3, 4))tensor([[[ 0.1815, -0.7603, 0.3465, 0.2593],[ 0.5841, 1.2581, 0.1583, -0.7283],[-0.0093, 0.6214, -0…
2022/1/23 6:07:52 人评论 次浏览
