mindspore.ops.Dropout
- class mindspore.ops.Dropout(keep_prob=0.5, Seed0=0, Seed1=0)[源代码]
Dropout是一种正则化手段,通过在训练中以 \(1 - keep\_prob\) 的概率随机将神经元输出设置为0,起到减少神经元相关性的作用,避免过拟合。
更多细节请参考
mindspore.ops.dropout()
。- 支持平台:
Ascend
GPU
CPU
样例:
>>> dropout = ops.Dropout3D(keep_prob=0.5) >>> x = Tensor(np.ones([1, 2, 3, 4, 5]), mindspore.float32) >>> output, mask = dropout(x) >>> print(output.shape) (1, 2, 3, 4, 5)