mlx.nn.Dropout

class Dropout(p: float = 0.5)

在训练期间随机将一部分元素归零。

剩余元素乘以 \(\frac{1}{1-p}\)，其中 \(p\) 是元素被归零的概率。这样做是为了使给定元素的期望值保持不变。

参数:

p (float) – 将元素归零的概率

方法