class Droupt: def __init__(self,dropout_ratio=0.5): self.dropout_ratio=dropout_ratio self.mask=None def forward(self,x,train_flg=True): if train_flg: self.mask=np.random.rand(*x.shape)>self.dropout_ratio return x*self.mask else: return x*(1.0-self.dropout_ratio) def backward(self,dout): return dout*self.mask

这个类是一个实现了dropout的神经网络层。在初始化时，可以指定dropout的比例，默认为0.5。在前向传播过程中，如果处于训练模式，会生成一个与输入x形状一致的掩码mask，其中元素值大于dropout比例的位置为True，小于等于dropout比例的位置为False。最后将输入x与掩码相乘，实现了随机丢弃一部分神经元的功能。如果处于测试模式，直接将输入x乘以（1.0 - dropout比例）即可。在反向传播过程中，将输出dout与掩码相乘，将被丢弃的神经元对应的梯度置为0。