pytorch圖

圖 計算圖 動態圖 計算圖 import torch w = torch.tensor([1.],requires_grad=True) x = torch.tensor([2.],requires_grad=True) a = torch.add(w,x) a.retain_grad() b = torch.add(w,1) y = torch.mul(a,b) y.backward()
相關文章
相關標籤/搜索