torch.randn不可导
时间: 2023-06-26 10:09:36 浏览: 66
0461-极智开发-解读torch.transpose的用法
是的,torch.randn()生成随机数的过程是不可导的。因此,如果在神经网络的反向传播中需要对随机数进行梯度计算,就不能使用torch.randn()函数。通常情况下,我们可以使用torch.nn.Parameter()将需要学习的参数初始化为随机数,并在反向传播中对其进行梯度计算。如果需要在模型中使用随机数,可以考虑使用torch.manual_seed()函数设置随机数种子,以确保结果的可重复性。
阅读全文