mindscience.models.layers.DropPath

class mindscience.models.layers.DropPath(dropout_rate=0.0)[源代码]

按样本进行路径丢弃（随机深度）（当应用于残差块的主路径时）。

参数：

dropout_rate (float, 可选) - 路径丢弃率，大于0且小于等于1。默认值：0.0。

输入：

x (Tensor) - 输入张量。

输出：

output (Tensor) - 输出张量，在训练期间应用了丢弃路径，在推理期间为原始输入。

样例：

>>> import mindspore as ms
>>> from mindspore import Tensor
>>> import mindspore.common.dtype as mstype
>>> import numpy as np
>>> from mindscience.models.layers import DropPath
>>> ms.set_context(mode=ms.GRAPH_MODE, save_graphs=False)
>>> x = Tensor(np.ones([2, 3, 4]), mstype.float32)
>>> droppath = DropPath(dropout_rate=0.1)
>>> output = droppath(x)
>>> print(output.shape)
(2, 3, 4)