![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
文章平均质量分 71
BigerBang
图像/视频修复与增强、超分辨率、sdr2hdr、插帧
展开
-
pytorch中的维度变换操作性质大总结:view, reshape, transpose, permute
在深度学习中,张量的维度变换是很重要的操作。在pytorch中,有四个用于维度变换的函数,view, reshape, transpose, permute。其中view和reshape都用于改变张量的形状,transpose, permute都用于重新排列张量的维度,但它们的功能和使用场景有所不同,下面将进行详细介绍,并给出测试验证代码,经过全面的了解,我们才能知道如何正确的使用这四个函数。原创 2024-06-06 21:03:24 · 1058 阅读 · 0 评论 -
pytorch保存和加载模型以及如何load部分参数
介绍pytorch模型的保存和加载,包括预训练模型和新定义的模型参数不完全一一对应的情况原创 2024-03-10 17:04:37 · 2703 阅读 · 0 评论 -
numpy、cupy、pytorch数组对象的相互转换
记录平常最常用的三个python对象之间的相互转换:numpy,cupy,pytorch三者的ndarray转换原创 2022-11-17 16:26:09 · 1826 阅读 · 0 评论 -
pycuda pytorch交互——使用pycuda处理pytorch tensor
很多时候,我们希望对tensor有一些自定义的操作,一种实现方式就是使用pycuda。本文以实现两个tensor的加法为例,讲解如何实现pycuda与pytorch交互。1. pycuda的使用方式首先看下pycuda文档对pycuda的定义:PyCUDA gives you easy, Pythonic access to Nvidia’s CUDA parallel computation API.即pycuda是在python中调用CUDA进行并行计算的接口。使用这个接口的流程为:1原创 2020-06-27 14:47:11 · 2203 阅读 · 8 评论 -
使用pytorch中的hook可视化feature map
利用pytorch中的钩子(hook),我们不用改变输入输出中间的网络结构,可以方便的获取、改变网络中间层变量的值和梯度。本篇文章记录了如何利用register_forward_hook(hook)获取pre-trained模型任意层的输出特征图并保存。register_forward_hook(hook)说明 Registers a forward hook on the module....原创 2019-08-04 15:12:29 · 5262 阅读 · 0 评论