mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
bfeff1eb8f
Summary: Closes https://github.com/pytorch/pytorch/issues/18724 Pull Request resolved: https://github.com/pytorch/pytorch/pull/19089 Differential Revision: D16073654 Pulled By: ezyang fbshipit-source-id: 5642179651ce45ab7c5a46cc1fcc4fd6b37fa71c
32 lines
640 B
Python
32 lines
640 B
Python
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: ...
|