Pytorch——conv2d參數使用

記錄下常用參數使用,方便查閱 conv2d的官方文檔點擊這裏 首先是原函數的參數: torch.nn.Conv2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True, padding_mode='zeros') 各個參數含義如下圖: 輸入輸出的格式: input
相關文章
相關標籤/搜索