torch/nn/modules/dropout.pyi.in

from ... import Tensor
from .module import Module


class _DropoutNd(Module):
    p: float
    inplace: bool

    def __init__(self, p: float = ..., inplace: bool = ...) -> None: ...

    def extra_repr(self): ...


class Dropout(_DropoutNd):
    def forward(self, input: Tensor) -> Tensor: ...


class Dropout2d(_DropoutNd):
    def forward(self, input: Tensor) -> Tensor: ...


class Dropout3d(_DropoutNd):
    def forward(self, input: Tensor) -> Tensor: ...


class AlphaDropout(_DropoutNd):
    def forward(self, input: Tensor) -> Tensor: ...


class FeatureAlphaDropout(_DropoutNd):
    def forward(self, input: Tensor) -> Tensor: ...