- 所有的tensor都有.requires_grad属性,可以设置这个属性.
x = tensor.ones(2,4,requires_grad=True)
2.如果想改变这个属性,就调用tensor.requires_grad_()方法:
x.requires_grad_(False/True)
参考文献 requires_grad_()与requires_grad的区别,同时pytorch的自动求导(AutoGrad)
PyTorch中tensor属性requires_grad_()与requires_grad
于 2022-04-07 19:27:43 首次发布