pytorch
framebreak
这个作者很懒,什么都没留下…
展开
-
PyTorch中 nn.Conv2d与nn.ConvTranspose2d函数的用法
通道数问题 :描述一个像素点,如果是灰度,那么只需要一个数值来描述它,就是单通道。如果有RGB三种颜色来描述它,就是三通道。最初输入的图片样本的 channels ,取决于图片类型;卷积操作完成后输出的 out_channels ,取决于卷积核的数量。此时的 out_channels 也会作为下一次卷积时的卷积核的 in_channels;卷积核中的 in_channels ,上面已经说...原创 2019-12-24 16:32:32 · 3452 阅读 · 1 评论 -
Pytorch——conv2d参数使用
torch.nn.Conv2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True, padding_mode='zeros')各个参数含义如下图:输入输出的格式:input(N,C-in,H-in,W-in)output(N,C-out,H-out,W...原创 2019-12-24 15:13:26 · 4574 阅读 · 0 评论 -
PyTorch学习之归一化层(BatchNorm、LayerNorm、InstanceNorm、GroupNorm)
BN,LN,IN,GN从学术化上解释差异:BatchNorm:batch方向做归一化,算NHW的均值,对小batchsize效果不好;BN主要缺点是对batchsize的大小比较敏感,由于每次计算均值和方差是在一个batch上,所以如果batchsize太小,则计算的均值、方差不足以代表整个数据分布LayerNorm:channel方向做归一化,算CHW的均值,主要对RNN作用明显;Inst...转载 2019-06-17 15:45:20 · 615 阅读 · 0 评论 -
PyTorch中permute的用法
permute(dims)将tensor的维度换位。参数:参数是一系列的整数,代表原来张量的维度。比如三维就有0,1,2这些dimension。例:import torchimport numpy as npa=np.array([[[1,2,3],[4,5,6]]])unpermuted=torch.tensor(a)print(unpermuted.size()) ...转载 2019-06-26 14:09:36 · 3530 阅读 · 0 评论