megengine.functional.dropout

dropout(inp, drop_prob, training=True)

返回一个新张量，其中每个元素按概率 P = drop_prob =``drop_prob`` 随机被设置为零。可以选择是否重新缩放输出张量。

参数

• inp (Tensor) – 输入张量。

• drop_prob (float) – 丢弃单个元素(将其设置为0)的概率

• training (bool) – dropout 在训练阶段的默认操作是 重新缩放输出张量, 这样它就可以在推理过程中被一个 Identity 代替, 默认值：True

返回类型

Tensor

返回

输出张量

例如：

import numpy as np
from megengine import tensor
import megengine.functional as F

x = tensor(np.ones(10, dtype=np.float32))
out = F.dropout(x, 1./3.)
print(out.numpy())

输出：

[1.5 1.5 0.  1.5 1.5 1.5 1.5 1.5 1.5 1.5]

megengine.functional.linear megengine.functional.one_hot