grad_fn=MeanBackward0 这是什么可不可以输出结果没有这个
这是PyTorch中的一个包含梯度信息的张量,表示该张量是由一系列操作计算得出的,并且可以反向传播梯度。如果想要输出结果没有这个grad_fn,可以使用detach()方法将其从计算图中分离出来,得到一个不包含梯度信息的新张量。例如:
import torch
x = torch.tensor([1.0, 2.0, 3.0], requires_grad=True)
y = x.mean()
print(y) # tensor(2., grad_fn=<MeanBackward0>)
y_detach = y.detach()
print(y_detach) # tensor(2.)
原文地址: https://www.cveoy.top/t/topic/eOtz 著作权归作者所有。请勿转载和采集!