megengine.functional.nn.pad

pad(src, pad_witdth, mode='constant', constant_value=0.0)[源代码]

Pad是megbrain padding算子的python包装, 可以填充任意一个维度, 维度最大为7。 pad模式默认为constant, 也支持edge(replicate) 和 reflect模式。

返回类型

Tensor