torch.nn.Module

相当于keras的Model和Layer,只不过pytorch不区分这两个。

当然实际上 keras的Model继承自network.Network,Network继承自Layer, Layer继承自module.Module。

Neural network module

nn.Module – Neural network module. Convenient way of encapsulating parameters, with helpers for moving them to GPU, exporting, loading, etc.

https://pytorch.org/tutorials/beginner/blitz/neural_networks_tutorial.html

以方便的方式封装参数,与帮助移动到GPU,导出,加载等。

上一页 -