Class FeatureAlphaDropoutImpl¶
Defined in File dropout.h
Page Contents
Inheritance Relationships¶
Base Type¶
public torch::nn::detail::_DropoutNd< FeatureAlphaDropoutImpl >(Template Class _DropoutNd)
Class Documentation¶
-
class FeatureAlphaDropoutImpl : public torch::nn::detail::_DropoutNd<FeatureAlphaDropoutImpl>¶
See the documentation for
torch::nn::FeatureAlphaDropoutOptionsclass to learn what constructor arguments are supported for this module.Example:
FeatureAlphaDropout model(FeatureAlphaDropoutOptions(0.2).inplace(true));
Public Functions
-
Tensor forward(const Tensor &input)¶
-
virtual void pretty_print(std::ostream &stream) const override¶
Pretty prints the
FeatureAlphaDropoutmodule into the givenstream.
-
Tensor forward(const Tensor &input)¶
