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)。
- 支持平台:
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]]