Class GELUImpl¶
Defined in File activation.h
Page Contents
Inheritance Relationships¶
Base Type¶
public torch::nn::Cloneable< GELUImpl >
(Template Class Cloneable)
Class Documentation¶
-
class
torch::nn
::
GELUImpl
: public torch::nn::Cloneable<GELUImpl>¶ Applies gelu over a given input.
See https://pytorch.org/docs/master/nn.html#torch.nn.GELU to learn about the exact behavior of this module.
Public Functions
-
GELUImpl
(GELUOptions options_ = {})¶
-
Tensor
forward
(const Tensor &input)¶
Public Members
-
GELUOptions
options
¶ The options with which this
Module
was constructed.
-