Linear(80,10)) @torch.jit.script_method def forward(self,x): return ... Used to get the output shape and dtype, No need to run the model ...
確定! 回上一頁