Class ThresholdImpl

Inheritance Relationships

Base Type

Class Documentation

class torch::nn::ThresholdImpl : public torch::nn::Cloneable<ThresholdImpl>

Applies the Threshold function element-wise.

See to learn about the exact behavior of this module.

See the documentation for torch::nn::ThresholdOptions class to learn what constructor arguments are supported for this module.


Threshold model(ThresholdOptions(42.42, 24.24).inplace(true));

Public Functions

ThresholdImpl(double threshold, double value)
ThresholdImpl(const ThresholdOptions &options_)
Tensor forward(Tensor input)
void reset() override

reset() must perform initialization of all members with reference semantics, most importantly parameters, buffers and submodules.

void pretty_print(std::ostream &stream) const override

Pretty prints the Threshold module into the given stream.

Public Members

ThresholdOptions options

The options with which this Module was constructed.


Access comprehensive developer documentation for PyTorch

View Docs


Get in-depth tutorials for beginners and advanced developers

View Tutorials


Find development resources and get your questions answered

View Resources