Class Hardshrink

• Defined in File activation.h

Inheritance Relationships

Base Type

• public torch::nn::ModuleHolder< HardshrinkImpl > (Template Class ModuleHolder)

Class Documentation

class Hardshrink : public torch::nn::ModuleHolder<HardshrinkImpl>

A ModuleHolder subclass for HardshrinkImpl.

See the documentation for HardshrinkImpl class to learn what methods it provides, and examples of how to use Hardshrink with torch::nn::HardshrinkOptions. See the documentation for ModuleHolder to learn about PyTorch’s module storage semantics.

Public Types

using Impl = HardshrinkImpl