Pytorch入門---學習筆記四---Autograd與邏輯迴歸

Autograd—自動求導系統 torch.autograd.backward 功能:自動求取梯度 tensors:用於求導的張量,如loss retain_graph:保存計算圖 create_graph:創建導數計算圖,用於高階求導 grad_tensors:多梯度權重
相關文章
相關標籤/搜索