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

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容