ReLU 差异对比

torch.nn.ReLU

torch.nn.ReLU(
   inplace=False
)

更多请查看 torch.nn.ReLU.

megengine.module.ReLU

megengine.module.ReLU(
   name=None
)

更多请查看 megengine.module.ReLU.

参数差异

inplace

Pytorch 中的 inplace 参数表示在不更改变量的内存地址的情况下，直接修改变量的值，MegEngine 无此参数。