mindspore/docs/api/api_python/ops/mindspore.ops.func_dropout.rst

22 lines
1.3 KiB
ReStructuredText
Raw Normal View History

2022-06-15 09:57:01 +08:00
mindspore.ops.dropout
======================
.. py:function:: mindspore.ops.dropout(x, p=0.5, seed0=0, seed1=0)
2022-06-15 09:57:01 +08:00
2022-11-28 11:30:29 +08:00
在训练期间,以服从伯努利分布的概率 `p` 随机将输入Tensor的某些值归零起到减少神经元相关性的作用避免过拟合。此概率与 `ops.dropout``nn.dropout` 中的含义相反。
2022-06-15 09:57:01 +08:00
2022-07-29 15:54:43 +08:00
参数:
- **x** (Tensor) - dropout的输入任意维度的Tensor其数据类型为float16或float32。
2022-11-01 19:52:13 +08:00
- **p** (float可选) - 输入神经元丢弃概率数值范围在0到1之间。例如p=0.1删除10%的神经元。默认值0.5。
- **seed0** (int可选) - 算子层的随机种子用于生成随机数。默认值0。
- **seed1** (int可选) - 全局的随机种子和算子层的随机种子共同决定最终生成的随机数。默认值0。
2022-07-29 15:54:43 +08:00
返回:
2022-11-28 11:30:29 +08:00
- **output** (Tensor) - 归零后的Tensorshape和数据类型与 `x` 相同。
- **mask** (Tensor) - 用于归零的掩码,内部会按位压缩与对齐。
2022-07-29 15:54:43 +08:00
异常:
- **TypeError** - `p` 不是float。
- **TypeError** - `seed0``seed1` 不是int。
2022-07-29 15:54:43 +08:00
- **TypeError** - `x` 的数据类型既不是float16也不是float32。
- **TypeError** - `x` 不是Tensor。