lstm

Documentation

Index

• func Reverse(x *gorgonia.Node, axis int) (*gorgonia.Node, error)
• type LSTMModule
  • func LSTM(m *godl.Model, opts LSTMOpts) *LSTMModule
  • func (m *LSTMModule) Forward(inputs ...*godl.Node) godl.Nodes
  • func (m *LSTMModule) Name() string
• type LSTMOpts
• type MergeMode

Functions

func Reverse

func Reverse(x *gorgonia.Node, axis int) (*gorgonia.Node, error)

Types

type LSTMModule

type LSTMModule struct {
	// contains filtered or unexported fields
}

func LSTM

func LSTM(m *godl.Model, opts LSTMOpts) *LSTMModule

func (*LSTMModule) Forward

func (m *LSTMModule) Forward(inputs ...*godl.Node) godl.Nodes

func (*LSTMModule) Name

func (m *LSTMModule) Name() string

type LSTMOpts

type LSTMOpts struct {
	InputDimension int
	HiddenSize     int
	// Layers         int
	Bidirectional bool
	MergeMode     MergeMode

	WithBias              bool
	WeightsInit, BiasInit gorgonia.InitWFn

	RecurrentActivation activation.Function
	Activation          activation.Function
}

type MergeMode

type MergeMode string

const (
	MergeModeConcat  MergeMode = "concat"
	MergeModeSum     MergeMode = "sum"
	MergeModeAverage MergeMode = "avg"
	MergeModeMul     MergeMode = "mul"
)