Parameter(torch.rand(3, 4)) self.linear = torch.nn.Linear(4, 5) def forward(self, ... GraphModule除了包含graph和code属性外就和正常的nn.
確定! 回上一頁