Pytorch – transpose和permute函数的区别和用法
1 transpose和permute函数的区别和用法 Pytorch中的transpose和permute方法都可以实现tensor维度之间的变换，transpose一次只能转换两个维度，而permute可以根据指定维度顺序一次转换多个维度，下面将简要介绍下transpose和permute方法。…
- Pytorch
- 2022-04-01
Pytorch – 使用torch.matmul()替换torch.einsum(‘nkctv,kvw->nctw’,(a,b))算子模式
在本文中，我将介绍如何使用普通算子matmul对torch.einsum('nkctv,kvw->nctw',(a,b))的算子模式进行替代。之前在另一篇文章：https://www.stubbornhuang.com/2065/中我已经详细介绍了pytorch的matmul方法，以及如何使用tor…
- Pytorch
- 2022-04-01
Pytorch – 使用torch.matmul()替换torch.einsum(‘bhxyd,md->bhxym’,(a,b))算子模式
在本文中，我将介绍如何使用普通算子matmul对torch.einsum('bhxyd,md->bhxym',(a,b))的算子模式进行替代。之前在另一篇文章：https://www.stubbornhuang.com/2065/中我已经详细介绍了pytorch的matmul方法，以及如何使用tor…
- Pytorch
- 2022-04-01
Pytorch – 使用torch.matmul()替换torch.einsum(‘nctw,cd->ndtw’,(a,b))算子模式
1 pytorch的torch.matmul()函数 函数形式 torch.matmul(input, other, *, out=None) → Tensor 该函数主要是用于求解两个tensor的矩阵乘积。 该函数根据输入的两个tensor的维度的不同进行不一样的张量运算，如下所示 如果两个参数…
- Pytorch
- 2022-03-29
Pytorch – 一文搞懂如何使用Pytorch构建与训练自定义深度学习网络（数据集自定义与加载，模型训练，模型测试，模型保存与加载）
使用Pytorch从零训练一个深度学习网络模型，常需要经过以下步骤：自定义数据集、加载自定义数据集、网络模型结构定义、定义损失函数、定义优化器、训练模型、测试模型、保存与加载模型等步骤。下文将详细阐述如何从零构建并训练一个深度学习网络模型的必要步骤。 1 自定义数据集 from torch.util…
- Pytorch
- 2021-12-22
Pytorch – 使用torch.onnx.export将Pytorch模型导出为ONNX模型
1 torch.onnx.export torch.onnx.export( model, args, f, export_params=True, verbose=False, training=<TrainingMode.EVAL: 0>, input_names=None, out…
- Pytorch
- 2021-09-16
Pytorch – 检测CUDA、cuDNN以及GPU版本的Pytorch是否安装成功、GPU显存测试
1 测试代码 import time import torch if __name__ == '__main__': print('torch版本：'+torch.__version__) print('cuda是否可用：'+str(torch.cuda.is_available())) print…
- Pytorch
- 2021-08-06
Pytorch – 使用opencv-python解码视频文件并将视频帧转换为Pytorch tensor作为网络模型输入数据
[toc] 1 视频文件作为网络模型的输入数据 越来越多的神经网络模型开始以视频作为训练数据，比如基于视频数据的行为识别等等，这就需要我们将视频转换为可适用的张量，本文将以pytorch为例，展示一下视频数据转换为pytorch tensor的过程。 2 使用OpenCV-Python解码视频文件并…
- Python
- 2021-04-19
Pytorch – 使用Pyav解码视频文件并将视频帧转换为Pytorch tensor作为网络模型输入数据
1 视频文件作为网络模型的输入数据 越来越多的神经网络模型开始以视频作为训练数据，比如基于视频数据的行为识别等等，这就需要我们将视频转换为可适用的张量，本文将以pytorch为例，展示一下视频数据转换为pytorch tensor的过程。 2 使用Pyav解码视频文件并将视频帧转换为Pytorch …
- Pytorch
- 2021-03-25