随笔分类 - pytorch学习
摘要：1. cuda和cudnn版本对应关系 https://developer.nvidia.com/rdp/cudnn archive a collapse742 10
阅读全文
摘要：torch.mul(a, b) 是矩阵 对应位相乘，即点乘操作， a和b的维度必须相等，a的维度是(1,2)， 则b的维度必须是（1,2）， 返回还是（1,2）的矩阵 torch.mm(a,b) 是矩阵a和b矩阵相乘，a的维度是（1,2），b的维度是（2,3），返回是（1,3）的矩阵 torch.b
阅读全文
摘要：1.查看数据类型 2.类型转换 方法一：简单后缀转换 方法二：使用torch.type()函数 方法三：使用type_as（tensor）将tensor转换为指定tensor的类型 3.tensor创建 指定维度和数据类型
阅读全文
摘要：1 list 与 numpy 2 numpy 与 tensor 3 list 与 tensor
阅读全文
摘要：1 torch.cat 将两个tensor在指定维度进行拼接 2 torch.stack 增加新的维度进行堆叠 3 torch.permute 调整tensor的维度顺序，相当于更灵活的transpose 4 tensor.contiguous view只能用在contiguous的tensor上。
阅读全文
摘要：1. nn.Conv2d 2. nn.BatchNorm2d 3. F.avg_pool2d
阅读全文
摘要：1.手动清空内存 查找占用显存的进程，然后kill 进程即可 2. 设置显卡
阅读全文
摘要：安装pytorch时需要cudatoolkit mkl pytorch。 每次使用在线安装时，都要重新从网上下载资源，速度较慢，遇到网络拥塞时易导致安装失败。把工具包在win10上通过迅雷等下载器下载，然后上传到服务器上，执行本地安装，以后重装环境时可以直接使用本地工具包安装，相对而言更快捷。如果需
阅读全文
摘要：``` import torch
import torch.nn as nn
import torch.nn.functional as F class ResidualBlock(nn.Module): ''' 实现子module: Residual Block ''' def __init__(self,inchannel,outcha...
阅读全文
