You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

mindspore.nn.Dropout.rst 1.6 kB

12345678910111213141516171819202122232425262728293031323334
  1. mindspore.nn.Dropout
  2. ====================
  3. .. py:class:: mindspore.nn.Dropout(keep_prob=0.5, dtype=mstype.float32)
  4. 随机丢弃层。
  5. Dropout是一种正则化手段,该算子根据丢弃概率 :math:`1 - keep\_prob` ,在训练过程中随机将一些神经元输出设置为0,通过阻止神经元节点间的相关性来减少过拟合,在推理过程中,此层返回与 `x` 相同的Tensor。
  6. 论文 `Dropout: A Simple Way to Prevent Neural Networks from Overfitting <http://www.cs.toronto.edu/~rsalakhu/papers/srivastava14a.pdf>`_ 中提出了该技术,并证明其能有效地减少过度拟合,防止神经元共适应。更多详细信息,请参见 `Improving neural networks by preventing co-adaptation of feature detectors <https://arxiv.org/pdf/1207.0580.pdf>`_ 。
  7. .. note::
  8. 训练过程中每步对同一通道(或神经元)独立进行丢弃。
  9. **参数:**
  10. - **keep_prob** (float) - 输入神经元保留率,数值范围在0到1之间。例如,rate=0.9,删除10%的神经元。默认值:0.5。
  11. - **dtype** (:class:`mindspore.dtype`) - `x` 的数据类型。默认值:float32。
  12. **输入:**
  13. - **x** (Tensor) - Dropout的输入,任意维度的Tensor。数据类型必须为float16或float32。
  14. **输出:**
  15. Tensor,输出为Tensor,其shape与 `x` shape相同。
  16. **异常:**
  17. - **TypeError** - `keep_prob` 不是浮点数。
  18. - **TypeError** - `x` 的dtype既不是float16也不是float32。
  19. - **ValueError** - `keep_prob` 不在范围(0, 1]内。
  20. - **ValueError** - `x` 的shape长度小于1。