torch.manual_seed
时间: 2024-01-15 18:05:40 浏览: 87
train.docx
torch.manual_seed是PyTorch中用于设置随机数种子的函数。在深度学习中,许多操作都涉及到随机数生成,如权重初始化、数据扰动等。为了确保模型的可重复性,我们需要设置随机数种子。使用torch.manual_seed函数可以设置PyTorch中所有随机数生成器的种子,从而使得随机数生成过程可重复。例如,可以使用以下代码设置随机数种子:
```
import torch
torch.manual_seed(42) # 设置随机数种子为42
```
这样,在接下来的模型训练中,使用的所有随机数都将是可预测的。
阅读全文