在PyTorch 中 torch.nn 專門用於實現神經網路。其中 nn.Module 包含了網路層的搭建,以及一個方法-- forward(input) ,並返回網路的輸出 outptu .
確定! 回上一頁