pytorch入門(1)

自動求導 import torch import numpy as np N, D_in, H, D_out = 64,1000,100,10 x = torch.randn(N,D_in) y = torch.randn(N,D_out) w1 = torch.randn(D_in, H,requires_grad = True) w2 = torch.randn(H,D_out,requ
相關文章
相關標籤/搜索