pytorch 自定義卷積核進行卷積操作

一 卷積操作:在pytorch搭建起網絡時,大家通常都使用已有的框架進行訓練,在網絡中使用最多就是卷積操作,最熟悉不過的就是 torch.nn.Conv2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True) 通過上面的輸入發現想自定義自己的卷積核,比如高斯核,發
相關文章
相關標籤/搜索