mindspore.ops.MirrorPad

class mindspore.ops.MirrorPad(mode='REFLECT')[源代码]

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

参数:
  • mode (str,可选) - 指定填充模式的可选字符串。可选值为: 'REFLECT''SYMMETRIC' 。默认值: 'REFLECT' 。 当采样grid超出输入Tensor的边界时,各种填充模式效果如下:

    • 'REFLECT' :使用零填充输入Tensor。例如,向 [1, 2, 3, 4] 的两边分别填充2个元素,结果为 [3, 2, 1, 2, 3, 4, 3, 2]。

    • 'SYMMETRIC' :使用Tensor边缘上像素的值填充输入Tensor。例如,向 [1, 2, 3, 4] 的两边分别填充2个元素,结果为 [2, 1, 1, 2, 3, 4, 4, 3]。

输入:
  • 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_xpadings 不是Tensor。

  • TypeError - mode 不是str。

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

支持平台:

Ascend 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]]