Shortcuts

Class ReLU6Impl

Inheritance Relationships

Base Type

Class Documentation

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

Applies the ReLU6 function element-wise.

See https://pytorch.org/docs/main/nn.html#torch.nn.ReLU6 to learn about the exact behavior of this module.

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

Example:

ReLU6 model(ReLU6Options().inplace(true));

Public Functions

explicit ReLU6Impl(const ReLU6Options &options_ = {})
Tensor forward(Tensor input)
virtual void reset() override

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

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

Pretty prints the ReLU6 module into the given stream.

Public Members

ReLU6Options options

The options with which this Module was constructed.

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources