當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


Python mxnet.gluon.nn.ReflectionPad2D用法及代碼示例


用法:

class mxnet.gluon.nn.ReflectionPad2D(padding=0, **kwargs)

參數

padding(int) - 整數填充大小

基礎:mxnet.gluon.block.HybridBlock

使用輸入邊界的反射填充輸入張量。

輸入:
  • data: 輸入張量的形狀\((N, C, H_{in}, W_{in})\) .
輸出:
  • out: 輸出具有形狀的張量\((N, C, H_{out}, W_{out})\) ,其中

例子

>>> m = nn.ReflectionPad2D(3)
>>> input = mx.nd.random.normal(shape=(16, 3, 224, 224))
>>> output = m(input)

相關用法


注:本文由純淨天空篩選整理自apache.org大神的英文原創作品 mxnet.gluon.nn.ReflectionPad2D。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。