MegEngine for PyTorch users

警告

Pytorch 的 Tensor 类中提供了许多操作/计算方法,而在 MegEngine 中这些方法被统一实现在 functional 模块中, 意味着类似 functional.add() 等操作并不一定存在着对应的 Tensor.add() 实现,这是设计上的历史决定。

警告

Pytorch 中默认所有 Tensor 都需要被求导,因此提供了 torch.no_grad 来禁用梯度计算。 而在 MegEngine 中 Tensor 默认不需要被求导,需要通过 megengine.autodiff.GradManager.attach 来进行绑定, 被绑定后的 Tensor 可以通过 megengine.Tensor.detach 来解除绑定。