layer

Documentation

Index

• type Layer
  • func NewLayer(config config.Config) *Layer
  • func (m *Layer) Forward(xs ...ag.Node) []ag.Node

Types

type Layer

type Layer struct {
	nn.BaseModel
	Config                 config.Config
	SelfAttention          *multiheadattention.Model
	SelfAttentionLayerNorm *layernorm.Model
	FFN                    *stack.Model
	LayerNorm              *layernorm.Model
}

Layer implements a BART encoder layer.

func NewLayer

func NewLayer(config config.Config) *Layer

NewLayer returns a new BART encoder Layer.

func (*Layer) Forward

func (m *Layer) Forward(xs ...ag.Node) []ag.Node

Forward performs the forward step for each input node and returns the result.