pytorch 求导

import torch
a = torch.ones([3, 4],requires_grad=True)
b = 3 
c = a * b
c.backward(torch.ones_like(c))
print(a.grad)