11 2021 档案

摘要:#计算x的梯度 import torch x = torch.arange(4.0) print(x) x.requires_grad_(True) #声明x需要计算梯度 print(x.grad) #默认值是None #定义y的值 y = 2 * torch.dot(x,x) print(y) y 阅读全文
posted @ 2021-11-12 11:19 小猪猪。。。 阅读(515) 评论(0) 推荐(0)