with torch.no

    科技2022-08-26  135

    #with torch.no_grad() 详解 ''' 被该语句包裹起来的语句将不会被追踪梯度 ''' a=torch.tensor([1.1],requires_grad=True) b=a*2 print(b)#tensor([2.2000], grad_fn=<MulBackward0>) b.add_(2) print(b)#tensor([4.2000], grad_fn=<AddBackward0>) with torch.no_grad(): b.mul_(2) print(b)#tensor([8.4000], grad_fn=<AddBackward0>) #被包裹的情况下,b.grad_fn还是add,说明mul并没有被追踪
    Processed: 0.017, SQL: 9