将# importtorchfromtorch.autogradimportvariable # tensor命名为a,指定变量中的值,并将tensor设定为可梯度下降。 在此,只有floattensor型tensor能够进行梯度下降a=variable(torch.FloatTensor ) [ 1,2 ]、[ 3,4 ] ),requires_grad=True 2]的4 )求出平方了a*a的平均值,代入out#,但由于后面的斜率下降,也就是说为了向a寻求指引,在这里将out作为a/4out=torch.mean(a*a ) ) d(out )/d ) a )=a/2打印(a.grad ) )结果,a/2=tensor ([ 0.5000,1.0000 ],# [ 1.5000,2.0000 ] )