栏目分类:
子分类:
返回
终身学习网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
终身学习网 > IT > 软件开发 > 后端开发 > Python

with torch.no

Python 更新时间:发布时间: 百科书网 趣学号

with torch.no_grad():

  • 停止对从跟踪历史中 的 .requires_grad=True 的张量自动求导。 参考这句话的网址
  • 举个例子:
  • import torch
    a = torch.randn(5,4,requires_grad=True)
    b = a*3
    with torch.no_grad():
        c = a*5
    print(a.requires_grad)
    print(b.requires_grad)
    print(c.requires_grad)
    # 输出结果:
    #True
    #True
    #False
    
    运行结果怎么来的:
    1.首先,a是随机生成了一个多维张量,并且设置了requires_grad属性为True,所以a.requires_grad的输出结果为TRUE是没有问题的。
    2.再来看b,b是由a通过运算得出的结果,所以b.requires_grad也是true没有问题。
    3.最后看c,c是在with torch.no_grad()的包裹中运算得出,它停止了跟踪,所以c.requires_grad为False
转载请注明:文章转载自 www.051e.com
本文地址:http://www.051e.com/it/294174.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 ©2023-2025 051e.com

ICP备案号:京ICP备12030808号