torch.nn.Module

相当于keras的Model和Layer,只不过pytorch不区分这两个。 当然实际上 keras的Mode … 继续阅读torch.nn.Module