ONNX算子:Pad

Pad[1]

输入:

  • data 需要填充的张量。
  • pads 整数张量,表示在待填充张量每个轴的开头和结尾需要添加或删除(如果为负)的填充元素的数量。
  • mode 填充模式。包含三种模式:constant reflect edge
  • constant_value 常量值(默认为0)。用于对输入张量进行填充。

输出:

  • output 填充后的张量。

constant padding

data =
[
    [1.0, 1.2],
    [2.3, 3.4],
    [4.5, 5.7],
]

pads = [0, 2, 0, 0]

mode = 'constant'

constant_value = 0.0

output =
[
    [0.0, 0.0, 1.0, 1.2],
    [0.0, 0.0, 2.3, 3.4],
    [0.0, 0.0, 4.5, 5.7],
]

reflect padding

pads = [0, 2, 0, 0]

mode = 'reflect'

output =
[
    [1.0, 1.2, 1.0, 1.2],
    [2.3, 3.4, 2.3, 3.4],
    [4.5, 5.7, 4.5, 5.7],
]

edge padding

pads = [0, 2, 0, 0]

mode = 'edge'

output =
[
    [1.0, 1.0, 1.0, 1.2],
    [2.3, 2.3, 2.3, 3.4],
    [4.5, 4.5, 4.5, 5.7],
]

  1. https://hub.fastgit.org/onnx/onnx/blob/master/docs/Operators.md#Pad ↩

你可能感兴趣的:(ONNX算子:Pad)