sample函数
时间: 2023-09-12 19:07:32 浏览: 151
sample功能列表说明1
sample函数通常用于从概率分布中随机采样,例如从一个概率分布中随机选择一个元素。在机器学习中,sample函数经常用于生成随机数,如在训练神经网络时,为了避免过拟合,我们会通过dropout的方式随机地丢弃一些神经元。下面是一个PyTorch中的例子,展示如何使用sample函数从正态分布中随机采样一个数:
```python
import torch
# 从正态分布中随机采样一个数
x = torch.randn(1)
# 打印采样结果
print(x.item())
```
这里我们使用了PyTorch中的randn函数生成了一个标准正态分布的随机数,然后使用了sample函数从这个分布中随机采样了一个数。需要注意的是,这里我们使用了item函数将张量转换为标量,方便打印结果。
阅读全文