encoder

Documentation

Index

• type Model
  • func New(config config.Config) *Model
  • func (m *Model) Encode(xs []ag.Node) []ag.Node

Types

type Model

type Model struct {
	nn.BaseModel
	Config             config.Config
	Layers             *stack.Model
	PositionalEncoder  positionalencoder.Encoder
	EmbeddingLayerNorm *layernorm.Model
	LayerNorm          *layernorm.Model
}

Model implements a BART encoder.

func New

func New(config config.Config) *Model

New returns a new BART encoder Model.

func (*Model) Encode

func (m *Model) Encode(xs []ag.Node) []ag.Node

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