pytorch ReplicationPad2d的張量流等效

Question

我正在嘗試找出如何在pytorch中執行以下填充的張量流等效項：

nn.ReplicationPad2d((1, 0, 1, 0))

我已經嘗試了以下方法，但是這僅在輸入張量實際上是2x2時才起作用：

tf.pad(my_tensor, [[1, 0], [1, 0]], "SYMMETRIC")

Answer 1

Tensorflow的等效項是tf.pad(my_tensor,[[0,0],[0,0],[1,0],[1,0]],"SYMMETRIC") 。 （這假定您有興趣對4D張量進行操作，前兩個維是批處理和通道）。

在Tensorflow中，您需要顯式給出所有四個維度的填充。 如果您不希望填充批處理和通道尺寸（在卷積網絡中通常不需要填充），則需要在張量的兩側顯式要求這兩個尺寸的零填充。 這就是為什么我在[1,0],[1,0]之前添加[0,0],[0,0] [1,0],[1,0] 。

在Pytorch，的一個實例nn.ReplicationPad2d已經假定為一個填充4D張量，沒有填充所述第一兩個維度。 這就是為什么您僅通過在兩個附加維度中指定填充來初始化實例。