Class AlphaDropoutImpl¶
Defined in File dropout.h
Page Contents
Inheritance Relationships¶
Base Type¶
public torch::nn::detail::_DropoutNd< AlphaDropoutImpl >
(Template Class _DropoutNd)
Class Documentation¶
-
class
torch::nn
::
AlphaDropoutImpl
: public torch::nn::detail::_DropoutNd<AlphaDropoutImpl>¶ Applies Alpha Dropout over the input.
See https://pytorch.org/docs/master/nn.html#torch.nn.AlphaDropout to learn about the exact behavior of this module.
See the documentation for
torch::nn::AlphaDropoutOptions
class to learn what constructor arguments are supported for this module.Example:
AlphaDropout model(AlphaDropoutOptions(0.2).inplace(true));
Public Functions
-
Tensor
forward
(const Tensor &input)¶
-
void
pretty_print
(std::ostream &stream) const override¶ Pretty prints the
AlphaDropout
module into the givenstream
.
-
Tensor