PyTorch的神經網絡模塊是torch.nn,它提供了用于構建神經網絡的模塊和函數。這個模塊包含了各種神經網絡層(如全連接層,卷積層,循環神經網絡層等),以及損失函數和優化器等工具,可以幫助用戶構建和訓練神經網絡模型。PyTorch的神經網絡模塊使得構建和訓練神經網絡變得更加簡單和方便。
torch.nn
億速云公眾號
手機網站二維碼
Copyright ? Yisu Cloud Ltd. All Rights Reserved. 2018 版權所有
廣州億速云計算有限公司粵ICP備17096448號-1 粵公網安備 44010402001142號增值電信業務經營許可證編號:B1-20181529