PyTorch中tensor属性requires_grad_()与requires_grad

  1. 所有的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)

你可能感兴趣的:(AI,深度学习,软件问题,python,pytorch)