![](https://img-blog.csdnimg.cn/20201014180756919.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
文章平均质量分 68
thinkpad1234567890
这个作者很懒,什么都没留下…
展开
-
关于torch.scatter函数
关于torch.scatter函数的理解原创 2022-06-06 16:34:58 · 368 阅读 · 0 评论 -
Pytorch中torch.gather函数
Pytorch中torch.gather函数解析转载 2022-06-06 16:13:57 · 269 阅读 · 0 评论 -
pytorch 卷积层到全连接层的参数个数
import torchfrom torch import nnfrom torch.autograd import Variableclass AlexNet(nn.Module): def __init__(self): super(AlexNet, self).__init__() self.conv_unit = nn.Sequentia...原创 2020-02-23 21:50:13 · 1816 阅读 · 0 评论 -
Ubuntu18.04+TITAN XP+anaconda+cuda10+cudnn+pytorch
开发环境:GPU: NVIDIA TITAN XPOS:Xubuntu 18.04新装系统,没有NVIDIA显卡驱动。先卸载已有驱动sudo apt-get remove --purge nvidia*安装库sudo apt-get updatesudo apt-get install dkms build-essential linux-headers-gener...原创 2020-02-07 22:56:53 · 1407 阅读 · 0 评论 -
pytorch torchvision.transforms.Normalize()中的mean和std参数---解惑
torchvision.transforms.Normalize()中的mean和std参数做什么用呢?疑问1:按照我的理解,归一化就是要把图片3个通道中的数据整理到[-1, 1]区间。x = (x - mean(x))/stddev(x)只要输入数据集x确定了,mean(x)和stddev(x)也就是确定的数值了,为什么Normalize()函数还需要输入mean和std的数值呢?疑...转载 2019-12-19 14:48:04 · 1948 阅读 · 1 评论