pytorch添加dropout

在 PyTorch 中，可以使用 `torch.nn.Dropout` 模块添加 dropout。具体实现方法如下：

import torch.nn as nn

# 创建一个dropout层，丢弃率为0.5
dropout = nn.Dropout(p=0.5)

# 将dropout层应用于输入张量
x = torch.randn(10, 20)  # 输入张量大小为10x20
x = dropout(x)

# 最后，可以将dropout层作为神经网络的一部分
model = nn.Sequential(
    nn.Linear(1024, 256),
    nn.ReLU(),
    nn.Dropout(p=0.5),
    nn.Linear(256, 10)
)

这样就创建了一个包含dropout的神经网络。