Shortcuts

Class GELUImpl

Inheritance Relationships

Base Type

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)
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 GELU module into the given stream.

Public Members

GELUOptions 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