layer

package
v1.0.2 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: May 8, 2025 License: MIT Imports: 7 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

This section is empty.

Types

type Attention added in v1.0.2 

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

func NewAttention added in v1.0.2 

func NewAttention(name string, dims, heads int, dropout float64, rope bool, opts ...LayerCreateOption) *Attention

func (*Attention) Args added in v1.0.2 

func (layer *Attention) Args() map[string]float32

func (*Attention) Class added in v1.0.2 

func (b *Attention) Class() string

func (*Attention) Clone added in v1.0.2 

func (layer *Attention) Clone() Layer

func (*Attention) Forward added in v1.0.2 

func (layer *Attention) Forward(q, k, v, mask *tensor.Tensor, isCausal, train bool) *tensor.Tensor

func (*Attention) Freeze added in v1.0.2 

func (layer *Attention) Freeze()

func (*Attention) Name added in v1.0.2 

func (b *Attention) Name() string

func (*Attention) Params added in v1.0.2 

func (layer *Attention) Params() []*tensor.Tensor

func (*Attention) Reset added in v1.0.2 

func (layer *Attention) Reset()

func (*Attention) Score added in v1.0.2 

func (layer *Attention) Score(q, k, v, mask *tensor.Tensor, isCausal, train bool) *tensor.Tensor

func (*Attention) SetROPEBase added in v1.0.2 

func (layer *Attention) SetROPEBase(n int64)

func (*Attention) ToScalarType added in v1.0.2 

func (layer *Attention) ToScalarType(t consts.ScalarType)

func (*Attention) Unfreeze added in v1.0.2 

func (layer *Attention) Unfreeze()

type Attention1 added in v1.0.2 

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

func NewAttention1 added in v1.0.2 

func NewAttention1(name string, dims, heads int, dropout float64, rope bool, opts ...LayerCreateOption) *Attention1

func (*Attention1) Args added in v1.0.2 

func (layer *Attention1) Args() map[string]float32

func (*Attention1) Class added in v1.0.2 

func (b *Attention1) Class() string

func (*Attention1) Clone added in v1.0.2 

func (layer *Attention1) Clone() Layer

func (*Attention1) Forward added in v1.0.2 

func (layer *Attention1) Forward(q, k, v, mask *tensor.Tensor, isCausal, train bool) *tensor.Tensor

func (*Attention1) Freeze added in v1.0.2 

func (layer *Attention1) Freeze()

func (*Attention1) Name added in v1.0.2 

func (b *Attention1) Name() string

func (*Attention1) Params added in v1.0.2 

func (layer *Attention1) Params() []*tensor.Tensor

func (*Attention1) Reset added in v1.0.2 

func (layer *Attention1) Reset()

func (*Attention1) Score added in v1.0.2 

func (layer *Attention1) Score(q, k, v, mask *tensor.Tensor, isCausal, train bool) *tensor.Tensor

func (*Attention1) SetROPEBase added in v1.0.2 

func (layer *Attention1) SetROPEBase(n int64)

func (*Attention1) ToScalarType added in v1.0.2 

func (layer *Attention1) ToScalarType(t consts.ScalarType)

func (*Attention1) Unfreeze added in v1.0.2 

func (layer *Attention1) Unfreeze()

type Conv1D added in v1.0.2 

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

func NewConv1D added in v1.0.2 

func NewConv1D(name string, inC, outC, kernel int, opts ...LayerCreateOption) *Conv1D

func (*Conv1D) Args added in v1.0.2 

func (layer *Conv1D) Args() map[string]float32

func (*Conv1D) Class added in v1.0.2 

func (b *Conv1D) Class() string

func (*Conv1D) Clone added in v1.0.2 

func (layer *Conv1D) Clone() Layer

func (*Conv1D) Forward added in v1.0.2 

func (layer *Conv1D) Forward(x *tensor.Tensor) *tensor.Tensor

func (*Conv1D) Freeze added in v1.0.2 

func (layer *Conv1D) Freeze()

func (*Conv1D) Name added in v1.0.2 

func (b *Conv1D) Name() string

func (*Conv1D) Params added in v1.0.2 

func (layer *Conv1D) Params() []*tensor.Tensor

func (*Conv1D) Reset added in v1.0.2 

func (layer *Conv1D) Reset()

func (*Conv1D) SetDilation added in v1.0.2 

func (layer *Conv1D) SetDilation(dilation int)

func (*Conv1D) SetGroups added in v1.0.2 

func (layer *Conv1D) SetGroups(groups int)

func (*Conv1D) SetPadding added in v1.0.2 

func (layer *Conv1D) SetPadding(padding int)

func (*Conv1D) SetStride added in v1.0.2 

func (layer *Conv1D) SetStride(stride int)

func (*Conv1D) ToScalarType added in v1.0.2 

func (layer *Conv1D) ToScalarType(t consts.ScalarType)

func (*Conv1D) Unfreeze added in v1.0.2 

func (layer *Conv1D) Unfreeze()

type Conv2D 

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

func NewConv2D 

func NewConv2D(name string, inC, outC int, kernel1, kernel2 int, opts ...LayerCreateOption) *Conv2D

func (*Conv2D) Args 

func (layer *Conv2D) Args() map[string]float32

func (*Conv2D) Class 

func (b *Conv2D) Class() string

func (*Conv2D) Clone added in v1.0.2 

func (layer *Conv2D) Clone() Layer

func (*Conv2D) Forward 

func (layer *Conv2D) Forward(x *tensor.Tensor) *tensor.Tensor

func (*Conv2D) Freeze added in v1.0.2 

func (layer *Conv2D) Freeze()

func (*Conv2D) Name 

func (b *Conv2D) Name() string

func (*Conv2D) Params 

func (layer *Conv2D) Params() []*tensor.Tensor

func (*Conv2D) Reset added in v1.0.2 

func (layer *Conv2D) Reset()

func (*Conv2D) SetDilation added in v1.0.2 

func (layer *Conv2D) SetDilation(dilation int)

func (*Conv2D) SetGroups added in v1.0.2 

func (layer *Conv2D) SetGroups(groups int)

func (*Conv2D) SetPadding added in v1.0.2 

func (layer *Conv2D) SetPadding(padding1, padding2 int)

func (*Conv2D) SetStride added in v1.0.2 

func (layer *Conv2D) SetStride(stride1, stride2 int)

func (*Conv2D) ToScalarType added in v1.0.2 

func (layer *Conv2D) ToScalarType(t consts.ScalarType)

func (*Conv2D) Unfreeze added in v1.0.2 

func (layer *Conv2D) Unfreeze()

type ConvTranspose1D added in v1.0.2 

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

func NewConvTranspose1D added in v1.0.2 

func NewConvTranspose1D(name string, inC, outC, kernel int, opts ...LayerCreateOption) *ConvTranspose1D

func (*ConvTranspose1D) Args added in v1.0.2 

func (layer *ConvTranspose1D) Args() map[string]float32

func (*ConvTranspose1D) Class added in v1.0.2 

func (b *ConvTranspose1D) Class() string

func (*ConvTranspose1D) Clone added in v1.0.2 

func (layer *ConvTranspose1D) Clone() Layer

func (*ConvTranspose1D) Forward added in v1.0.2 

func (layer *ConvTranspose1D) Forward(x *tensor.Tensor) *tensor.Tensor

func (*ConvTranspose1D) Freeze added in v1.0.2 

func (layer *ConvTranspose1D) Freeze()

func (*ConvTranspose1D) Name added in v1.0.2 

func (b *ConvTranspose1D) Name() string

func (*ConvTranspose1D) Params added in v1.0.2 

func (layer *ConvTranspose1D) Params() []*tensor.Tensor

func (*ConvTranspose1D) Reset added in v1.0.2 

func (layer *ConvTranspose1D) Reset()

func (*ConvTranspose1D) SetDilation added in v1.0.2 

func (layer *ConvTranspose1D) SetDilation(dilation int)

func (*ConvTranspose1D) SetGroups added in v1.0.2 

func (layer *ConvTranspose1D) SetGroups(groups int)

func (*ConvTranspose1D) SetOutputPadding added in v1.0.2 

func (layer *ConvTranspose1D) SetOutputPadding(padding int)

func (*ConvTranspose1D) SetPadding added in v1.0.2 

func (layer *ConvTranspose1D) SetPadding(padding int)

func (*ConvTranspose1D) SetStride added in v1.0.2 

func (layer *ConvTranspose1D) SetStride(stride int)

func (*ConvTranspose1D) ToScalarType added in v1.0.2 

func (layer *ConvTranspose1D) ToScalarType(t consts.ScalarType)

func (*ConvTranspose1D) Unfreeze added in v1.0.2 

func (layer *ConvTranspose1D) Unfreeze()

type ConvTranspose2D added in v1.0.2 

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

func NewConvTranspose2D added in v1.0.2 

func NewConvTranspose2D(name string, inC, outC int, kernel1, kernel2 int, opts ...LayerCreateOption) *ConvTranspose2D

func (*ConvTranspose2D) Args added in v1.0.2 

func (layer *ConvTranspose2D) Args() map[string]float32

func (*ConvTranspose2D) Class added in v1.0.2 

func (b *ConvTranspose2D) Class() string

func (*ConvTranspose2D) Clone added in v1.0.2 

func (layer *ConvTranspose2D) Clone() Layer

func (*ConvTranspose2D) Forward added in v1.0.2 

func (layer *ConvTranspose2D) Forward(x *tensor.Tensor) *tensor.Tensor

func (*ConvTranspose2D) Freeze added in v1.0.2 

func (layer *ConvTranspose2D) Freeze()

func (*ConvTranspose2D) Name added in v1.0.2 

func (b *ConvTranspose2D) Name() string

func (*ConvTranspose2D) Params added in v1.0.2 

func (layer *ConvTranspose2D) Params() []*tensor.Tensor

func (*ConvTranspose2D) Reset added in v1.0.2 

func (layer *ConvTranspose2D) Reset()

func (*ConvTranspose2D) SetDilation added in v1.0.2 

func (layer *ConvTranspose2D) SetDilation(dilation int)

func (*ConvTranspose2D) SetGroups added in v1.0.2 

func (layer *ConvTranspose2D) SetGroups(groups int)

func (*ConvTranspose2D) SetOutputPadding added in v1.0.2 

func (layer *ConvTranspose2D) SetOutputPadding(padding1, padding2 int)

func (*ConvTranspose2D) SetPadding added in v1.0.2 

func (layer *ConvTranspose2D) SetPadding(padding1, padding2 int)

func (*ConvTranspose2D) SetStride added in v1.0.2 

func (layer *ConvTranspose2D) SetStride(stride1, stride2 int)

func (*ConvTranspose2D) ToScalarType added in v1.0.2 

func (layer *ConvTranspose2D) ToScalarType(t consts.ScalarType)

func (*ConvTranspose2D) Unfreeze added in v1.0.2 

func (layer *ConvTranspose2D) Unfreeze()

type Dropout 

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

func NewDropout 

func NewDropout(name string, keep float64) *Dropout

func (*Dropout) Args 

func (layer *Dropout) Args() map[string]float32

func (*Dropout) Class 

func (b *Dropout) Class() string

func (*Dropout) Clone added in v1.0.2 

func (layer *Dropout) Clone() Layer

func (*Dropout) Forward 

func (layer *Dropout) Forward(x *tensor.Tensor, train bool) *tensor.Tensor

func (*Dropout) Freeze added in v1.0.2 

func (b *Dropout) Freeze()

func (*Dropout) Name 

func (b *Dropout) Name() string

func (*Dropout) Params 

func (b *Dropout) Params() []*tensor.Tensor

func (*Dropout) Reset added in v1.0.2 

func (layer *Dropout) Reset()

func (*Dropout) ToScalarType added in v1.0.2 

func (layer *Dropout) ToScalarType(t consts.ScalarType)

func (*Dropout) Unfreeze added in v1.0.2 

func (b *Dropout) Unfreeze()

type Embedding added in v1.0.2 

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

func NewEmbedding added in v1.0.2 

func NewEmbedding(name string, num, dim int, opts ...LayerCreateOption) *Embedding

func (*Embedding) Args added in v1.0.2 

func (layer *Embedding) Args() map[string]float32

func (*Embedding) Class added in v1.0.2 

func (b *Embedding) Class() string

func (*Embedding) Clone added in v1.0.2 

func (layer *Embedding) Clone() Layer

func (*Embedding) Forward added in v1.0.2 

func (layer *Embedding) Forward(x *tensor.Tensor) *tensor.Tensor

func (*Embedding) Freeze added in v1.0.2 

func (layer *Embedding) Freeze()

func (*Embedding) Name added in v1.0.2 

func (b *Embedding) Name() string

func (*Embedding) Params added in v1.0.2 

func (layer *Embedding) Params() []*tensor.Tensor

func (*Embedding) Reset added in v1.0.2 

func (layer *Embedding) Reset()

func (*Embedding) SetPaddingIdx added in v1.0.2 

func (layer *Embedding) SetPaddingIdx(n int64)

func (*Embedding) ToScalarType added in v1.0.2 

func (layer *Embedding) ToScalarType(t consts.ScalarType)

func (*Embedding) Unfreeze added in v1.0.2 

func (layer *Embedding) Unfreeze()

type Flatten added in v1.0.0 

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

func NewFlatten added in v1.0.0 

func NewFlatten(name string) *Flatten

func (*Flatten) Args added in v1.0.0 

func (b *Flatten) Args() map[string]float32

func (*Flatten) Class added in v1.0.0 

func (b *Flatten) Class() string

func (*Flatten) Clone added in v1.0.2 

func (layer *Flatten) Clone() Layer

func (*Flatten) Forward added in v1.0.0 

func (layer *Flatten) Forward(x *tensor.Tensor) *tensor.Tensor

func (*Flatten) Freeze added in v1.0.2 

func (b *Flatten) Freeze()

func (*Flatten) Name added in v1.0.0 

func (b *Flatten) Name() string

func (*Flatten) Params added in v1.0.0 

func (b *Flatten) Params() []*tensor.Tensor

func (*Flatten) Reset added in v1.0.2 

func (layer *Flatten) Reset()

func (*Flatten) ToScalarType added in v1.0.2 

func (layer *Flatten) ToScalarType(t consts.ScalarType)

func (*Flatten) Unfreeze added in v1.0.2 

func (b *Flatten) Unfreeze()

type Layer 

type Layer interface {
	Params() []*tensor.Tensor
	Class() string
	Name() string
	Args() map[string]float32
	Freeze()
	Unfreeze()
	ToScalarType(t consts.ScalarType)
	Reset()
	Clone() Layer
}

func LoadAttention added in v1.0.2 

func LoadAttention(name string, params []*tensor.Tensor, args map[string]float32) Layer

func LoadAttention1 added in v1.0.2 

func LoadAttention1(name string, params []*tensor.Tensor, args map[string]float32) Layer

func LoadConv1D added in v1.0.2 

func LoadConv1D(name string, params []*tensor.Tensor, args map[string]float32) Layer

func LoadConv2D 

func LoadConv2D(name string, params []*tensor.Tensor, args map[string]float32) Layer

func LoadConvTranspose1D added in v1.0.2 

func LoadConvTranspose1D(name string, params []*tensor.Tensor, args map[string]float32) Layer

func LoadConvTranspose2D added in v1.0.2 

func LoadConvTranspose2D(name string, params []*tensor.Tensor, args map[string]float32) Layer

func LoadDropout 

func LoadDropout(name string, _ []*tensor.Tensor, args map[string]float32) Layer

func LoadEmbedding added in v1.0.2 

func LoadEmbedding(name string, params []*tensor.Tensor, args map[string]float32) Layer

func LoadFlatten added in v1.0.0 

func LoadFlatten(name string, _ []*tensor.Tensor, _ map[string]float32) Layer

func LoadLayerNorm added in v1.0.2 

func LoadLayerNorm(name string, params []*tensor.Tensor, args map[string]float32) Layer

func LoadLinear added in v1.0.2 

func LoadLinear(name string, params []*tensor.Tensor, args map[string]float32) Layer

func LoadLstm added in v1.0.0 

func LoadLstm(name string, params []*tensor.Tensor, args map[string]float32) Layer

func LoadMaxPool1D added in v1.0.2 

func LoadMaxPool1D(name string, _ []*tensor.Tensor, args map[string]float32) Layer

func LoadRMSNorm added in v1.0.2 

func LoadRMSNorm(name string, params []*tensor.Tensor, args map[string]float32) Layer

func LoadReZero added in v1.0.2 

func LoadReZero(name string, params []*tensor.Tensor, args map[string]float32) Layer

func LoadRnn 

func LoadRnn(name string, params []*tensor.Tensor, args map[string]float32) Layer

type LayerCreateOption added in v1.0.2 

type LayerCreateOption func(*base)

func WithDevice added in v1.0.2 

func WithDevice(device consts.DeviceType) LayerCreateOption

func WithInitializer added in v1.0.2 

func WithInitializer(init initializer.Initializer) LayerCreateOption

func WithParamType added in v1.0.2 

func WithParamType(t consts.ScalarType) LayerCreateOption

type LayerNorm added in v1.0.2 

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

func NewLayerNorm added in v1.0.2 

func NewLayerNorm(name string, dims int64, opts ...LayerCreateOption) *LayerNorm

func (*LayerNorm) Args added in v1.0.2 

func (b *LayerNorm) Args() map[string]float32

func (*LayerNorm) Class added in v1.0.2 

func (b *LayerNorm) Class() string

func (*LayerNorm) Clone added in v1.0.2 

func (layer *LayerNorm) Clone() Layer

func (*LayerNorm) Forward added in v1.0.2 

func (layer *LayerNorm) Forward(x *tensor.Tensor) *tensor.Tensor

func (*LayerNorm) Freeze added in v1.0.2 

func (layer *LayerNorm) Freeze()

func (*LayerNorm) Name added in v1.0.2 

func (b *LayerNorm) Name() string

func (*LayerNorm) Params added in v1.0.2 

func (layer *LayerNorm) Params() []*tensor.Tensor

func (*LayerNorm) Reset added in v1.0.2 

func (layer *LayerNorm) Reset()

func (*LayerNorm) ToScalarType added in v1.0.2 

func (layer *LayerNorm) ToScalarType(t consts.ScalarType)

func (*LayerNorm) Unfreeze added in v1.0.2 

func (layer *LayerNorm) Unfreeze()

type Linear added in v1.0.2 

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

func NewLinear added in v1.0.2 

func NewLinear(name string, input, output int, opts ...LayerCreateOption) *Linear

func (*Linear) Args added in v1.0.2 

func (layer *Linear) Args() map[string]float32

func (*Linear) Class added in v1.0.2 

func (b *Linear) Class() string

func (*Linear) Clone added in v1.0.2 

func (layer *Linear) Clone() Layer

func (*Linear) Forward added in v1.0.2 

func (layer *Linear) Forward(x *tensor.Tensor) *tensor.Tensor

func (*Linear) Freeze added in v1.0.2 

func (layer *Linear) Freeze()

func (*Linear) Name added in v1.0.2 

func (b *Linear) Name() string

func (*Linear) Params added in v1.0.2 

func (layer *Linear) Params() []*tensor.Tensor

func (*Linear) Reset added in v1.0.2 

func (layer *Linear) Reset()

func (*Linear) ToScalarType added in v1.0.2 

func (layer *Linear) ToScalarType(t consts.ScalarType)

func (*Linear) Unfreeze added in v1.0.2 

func (layer *Linear) Unfreeze()

type Lstm added in v1.0.0 

type Lstm struct {
	Wi, Bi *tensor.Tensor
	Wf, Bf *tensor.Tensor
	Wg, Bg *tensor.Tensor
	Wo, Bo *tensor.Tensor
	// contains filtered or unexported fields
}

func NewLstm added in v1.0.0 

func NewLstm(name string, featureSize, steps, hidden int, opts ...LayerCreateOption) *Lstm

func (*Lstm) Args added in v1.0.0 

func (layer *Lstm) Args() map[string]float32

func (*Lstm) Class added in v1.0.0 

func (b *Lstm) Class() string

func (*Lstm) Clone added in v1.0.2 

func (layer *Lstm) Clone() Layer

func (*Lstm) Forward added in v1.0.0 

func (layer *Lstm) Forward(x, h, c *tensor.Tensor) (*tensor.Tensor, *tensor.Tensor, *tensor.Tensor)

func (*Lstm) Freeze added in v1.0.2 

func (layer *Lstm) Freeze()

func (*Lstm) Name added in v1.0.0 

func (b *Lstm) Name() string

func (*Lstm) Params added in v1.0.0 

func (layer *Lstm) Params() []*tensor.Tensor

func (*Lstm) Reset added in v1.0.2 

func (layer *Lstm) Reset()

func (*Lstm) ToScalarType added in v1.0.2 

func (layer *Lstm) ToScalarType(t consts.ScalarType)

func (*Lstm) Unfreeze added in v1.0.2 

func (layer *Lstm) Unfreeze()

type MaxPool1D added in v1.0.2 

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

func NewMaxPool1D added in v1.0.2 

func NewMaxPool1D(name string, kernel int, opts ...LayerCreateOption) *MaxPool1D

func (*MaxPool1D) Args added in v1.0.2 

func (layer *MaxPool1D) Args() map[string]float32

func (*MaxPool1D) Class added in v1.0.2 

func (b *MaxPool1D) Class() string

func (*MaxPool1D) Clone added in v1.0.2 

func (layer *MaxPool1D) Clone() Layer

func (*MaxPool1D) Forward added in v1.0.2 

func (layer *MaxPool1D) Forward(x *tensor.Tensor) *tensor.Tensor

func (*MaxPool1D) Freeze added in v1.0.2 

func (b *MaxPool1D) Freeze()

func (*MaxPool1D) Name added in v1.0.2 

func (b *MaxPool1D) Name() string

func (*MaxPool1D) Params added in v1.0.2 

func (b *MaxPool1D) Params() []*tensor.Tensor

func (*MaxPool1D) Reset added in v1.0.2 

func (layer *MaxPool1D) Reset()

func (*MaxPool1D) SetCeil added in v1.0.2 

func (layer *MaxPool1D) SetCeil(ceil bool)

func (*MaxPool1D) SetDilation added in v1.0.2 

func (layer *MaxPool1D) SetDilation(dilation int)

func (*MaxPool1D) SetPadding added in v1.0.2 

func (layer *MaxPool1D) SetPadding(padding int)

func (*MaxPool1D) SetStride added in v1.0.2 

func (layer *MaxPool1D) SetStride(stride int)

func (*MaxPool1D) ToScalarType added in v1.0.2 

func (layer *MaxPool1D) ToScalarType(t consts.ScalarType)

func (*MaxPool1D) Unfreeze added in v1.0.2 

func (b *MaxPool1D) Unfreeze()

type RMSNorm added in v1.0.2 

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

func NewRMSNorm added in v1.0.2 

func NewRMSNorm(name string, dims int64, opts ...LayerCreateOption) *RMSNorm

func (*RMSNorm) Args added in v1.0.2 

func (b *RMSNorm) Args() map[string]float32

func (*RMSNorm) Class added in v1.0.2 

func (b *RMSNorm) Class() string

func (*RMSNorm) Clone added in v1.0.2 

func (layer *RMSNorm) Clone() Layer

func (*RMSNorm) Forward added in v1.0.2 

func (layer *RMSNorm) Forward(x *tensor.Tensor) *tensor.Tensor

func (*RMSNorm) Freeze added in v1.0.2 

func (layer *RMSNorm) Freeze()

func (*RMSNorm) Name added in v1.0.2 

func (b *RMSNorm) Name() string

func (*RMSNorm) Params added in v1.0.2 

func (layer *RMSNorm) Params() []*tensor.Tensor

func (*RMSNorm) Reset added in v1.0.2 

func (layer *RMSNorm) Reset()

func (*RMSNorm) ToScalarType added in v1.0.2 

func (layer *RMSNorm) ToScalarType(t consts.ScalarType)

func (*RMSNorm) Unfreeze added in v1.0.2 

func (layer *RMSNorm) Unfreeze()

type ReZero added in v1.0.2 

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

func NewReZero added in v1.0.2 

func NewReZero(name string, opts ...LayerCreateOption) *ReZero

func (*ReZero) Args added in v1.0.2 

func (layer *ReZero) Args() map[string]float32

func (*ReZero) Class added in v1.0.2 

func (b *ReZero) Class() string

func (*ReZero) Clone added in v1.0.2 

func (layer *ReZero) Clone() Layer

func (*ReZero) Forward added in v1.0.2 

func (layer *ReZero) Forward(x *tensor.Tensor) *tensor.Tensor

func (*ReZero) Freeze added in v1.0.2 

func (layer *ReZero) Freeze()

func (*ReZero) Name added in v1.0.2 

func (b *ReZero) Name() string

func (*ReZero) Params added in v1.0.2 

func (layer *ReZero) Params() []*tensor.Tensor

func (*ReZero) Reset added in v1.0.2 

func (layer *ReZero) Reset()

func (*ReZero) ToScalarType added in v1.0.2 

func (layer *ReZero) ToScalarType(t consts.ScalarType)

func (*ReZero) Unfreeze added in v1.0.2 

func (layer *ReZero) Unfreeze()

type Rnn 

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

func NewRnn 

func NewRnn(name string, featureSize, steps, hidden int, opts ...LayerCreateOption) *Rnn

func (*Rnn) Args 

func (layer *Rnn) Args() map[string]float32

func (*Rnn) Class 

func (b *Rnn) Class() string

func (*Rnn) Clone added in v1.0.2 

func (layer *Rnn) Clone() Layer

func (*Rnn) Forward 

func (layer *Rnn) Forward(x, h *tensor.Tensor) (*tensor.Tensor, *tensor.Tensor)

func (*Rnn) Freeze added in v1.0.2 

func (layer *Rnn) Freeze()

func (*Rnn) Name 

func (b *Rnn) Name() string

func (*Rnn) Params 

func (layer *Rnn) Params() []*tensor.Tensor

func (*Rnn) Reset added in v1.0.2 

func (layer *Rnn) Reset()

func (*Rnn) ToScalarType added in v1.0.2 

func (layer *Rnn) ToScalarType(t consts.ScalarType)

func (*Rnn) Unfreeze added in v1.0.2 

func (layer *Rnn) Unfreeze()

Source Files

View all Source files

Directories

Path Synopsis

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.