Class TanhshrinkImpl#
Defined in File activation.h
Page Contents
Inheritance Relationships#
Base Type#
public torch::nn::Cloneable< TanhshrinkImpl >(Template Class Cloneable)
Class Documentation#
-
class TanhshrinkImpl : public torch::nn::Cloneable<TanhshrinkImpl>#
Applies Tanhshrink over a given input.
See https://pytorch.org/docs/main/nn.html#torch.nn.Tanhshrink to learn about the exact behavior of this module.
Public Functions
-
Tensor forward(const 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
Tanhshrinkmodule into the givenstream.
-
Tensor forward(const Tensor &input)#