哈尔滨工业大学
pytorch入门
墨衍会员 · AI 创作全网分发 墨衍智能分发 通过墨衍一键同步至各平台 可同步平台 CSDN微信公众号微博知乎掘金百家号博客园抖音小红书 你的文章也可以这样分发 墨衍会员 ¥399起/年，写一次发全网
我也要 关注 + 加入社区pytorch入门
import torchdef forward(x): return x*wdef loss(x, y): y_hat = forward(x) return (y-y_hat)**2if __name__ == "__main__": x_data = [1., 2., 3.] y_data = [2., 4., 6.] w = torch.tensor([1.0]) w.requires_grad = True for
