pytorch零填充函數ZeroPad2d作用

pytorch中最常用的零填充函數是nn.ZeroPad2d,也就是對Tensor使用0進行邊界填充,我們可以指定tensor的四個方向上的填充數,比如左邊添加1dim、右邊添加2dim、上邊添加3dim、下邊添加4dim,即指定paddin參數爲(1,2,3,4),如下: pad = nn.ZeroPad2d(padding=(1, 2, 3, 4)) y = pad(x) 得到的y是x在四個方
相關文章
相關標籤/搜索