LeakyRelu 差异对比

torch.nn.LeakyRelu

torch.nn.LeakyRelu(
   negative_slope=0.01,
   inplace=False
)

更多请查看 torch.nn.LeakyRelu.

megengine.module.LeakyRelu

megengine.module.LeakyRelu(
   negative_slope=0.01,
   ** kwargs
)

更多请查看 megengine.module.LeakyRelu.

参数差异

inplace

Pytorch 中的 inplace 参数表示是否要原地操作，默认不开启，MegEngine 无此参数。