DEV Community
·
PyTorch中的Dropout()
💡
原文英文，约400词，阅读约需2分钟。
📝
内容提要
Dropout Layer是一个随机将张量中的元素置零或乘以一个值的操作。可以通过设置“p”参数来确定元素被置零的概率。通过设置“inplace”参数，可以决定操作是否原地执行。文章提供了使用不同张量的Dropout Layer的示例。
🎯
关键要点
- Dropout Layer是一个随机将张量中的元素置零或乘以一个值的操作。
- 通过设置'p'参数可以确定元素被置零的概率，范围为0到1。
- 通过设置'inplace'参数可以决定操作是否原地执行，建议保持为False。
- Dropout Layer的输入是一个浮点数张量，默认情况下'requires_grad'为False。
- 示例中展示了使用不同张量的Dropout Layer，包括一维、二维和三维张量。
- 使用不同的'p'值可以观察到不同的输出结果。
➡️
