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