首页
科技
登录
6mi
u
盘
搜
搜 索
科技
with torch.no
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并没有被追踪
转载请注明原文地址:https://blackberry.8miu.com/read-17568.html
最新回复
(
0
)