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
相关文章
相关标签/搜索