技术栈
autograd
伊织code
15 小时前
pytorch
·
api
·
cpu
·
cuda
·
微分
·
autograd
PyTorch API 2
torch.autograd 提供了实现任意标量值函数自动微分的类和函数。只需对现有代码进行最小改动——您只需要通过requires_grad=True关键字声明需要计算梯度的Tensor即可。目前，我们仅支持浮点型Tensor(包括half、float、double和bfloat16)和复数型Tensor(cfloat、cdouble)的自动微分功能。