mindspore.ops.MirrorPad

class mindspore.ops.MirrorPad(mode='REFLECT')

通过指定的填充模式和大小对输入Tensor进行填充。

参数：

• mode (str) - 指定填充模式。可选值：”REFLECT”和”SYMMETRIC”。 默认值：”REFLECT”。

输入：

• input_x (Tensor) - shape: \((N, *)\) ，其中 \(*\) 表示任何数量的附加维度。

• paddings (Tensor) - shape为 \((N, 2)\) 的矩阵。N为输入Tensor的秩。int类型。 对于输入的第 D 个维度，paddings[D, 0] 表示需在输入第 D 维头部填充的数量，paddings[D, 1] 表示需在输入第 D 维尾部填充的数量。

输出：

填充后的Tensor。

• 如果设置 mode 为”REFLECT”，将使用对称轴对称复制的方式来进行填充。 如果 input_x 为[[1,2,3]，[4,5,6]，[7,8,9]]， paddings 为[[1,1], [2,2]]，则输出为[[6,5,4,5,6,5,4]，[3,2,1,2,3,2,1]，[6,5,4,5,6,5,4]，[9,8,7,8,9,8,7]，[6,5,4,5,6,5,4]]。 更直观的理解请参见下面的样例。

• 如果 mode 为”SYMMETRIC”，则填充方法类似于”REFLECT”。它也会根据对称轴复制，但是也包括对称轴。如果 input_x 为[[1,2,3],[4,5,6],[7,8,9]]， paddings 为[[1,1], [2,2]]，则输出为[[2,1,1,2,3,3,2]，[2,1,1,2,3,3,2]，[5,4,4,5,6,6,5]，[8,7,7,8,9,9,8]，[8,7,7,8,9,9,8]]。 更直观的理解请参见下面的样例。

异常：

• TypeError - input_x 或 padings 不是Tensor。

• TypeError - mode 不是str。

• ValueError - paddings.size 不等于2 * len(input_x)。

支持平台：

GPU CPU

样例：

>>> from mindspore import Tensor, nn, ops
>>> # case1: mode="REFLECT"
>>> class Net(nn.Cell):
...    def __init__(self, mode):
...        super(Net, self).__init__()
...        self.pad = ops.MirrorPad(mode=mode)
...        self.paddings = Tensor([[1, 1], [2, 2]])
...    def construct(self, input_x):
...        return self.pad(input_x, self.paddings)
...
>>> input_x = Tensor([[1,2,3], [4,5,6], [7,8,9]])
>>> pad = Net("REFLECT")
>>> output = pad(input_x)
>>> print(output)
[[6 5 4 5 6 5 4]
 [3 2 1 2 3 2 1]
 [6 5 4 5 6 5 4]
 [9 8 7 8 9 8 7]
 [6 5 4 5 6 5 4]]
>>> # case2: mode="SYMMETRIC"
>>> pad = Net("SYMMETRIC")
>>> output = pad(input_x)
>>> print(output)
[[2 1 1 2 3 3 2]
 [2 1 1 2 3 3 2]
 [5 4 4 5 6 6 5]
 [8 7 7 8 9 9 8]
 [8 7 7 8 9 9 8]]