Contents

struct

SushiAI::NN::GELU

The Gaussian error linear unit, tanh approximation.

Declared in
include/SushiAI/nn/layers.hpp

Static public member functions

static constexpr auto fields()

No parameters.

Public member functions

Graph::ValueId forward(const Graph::GraphBuilder &builder, Graph::ValueId x) const

Traces the activation.

Parameters

builder

Where the node goes.

x

The input.

Returns

The result's id.

Shape output_shape(const Shape &input) const

Returns the per-sample shape forward() produces, which is input.