layer

Documentation

Index

• type ConvTranspose2dLayer
  • func NewConvTranspose2dLayer(inChannels, outChannels, kernelSizeRows, kernelSizeCols, strideRows, ... int) *ConvTranspose2dLayer
  • func (ct *ConvTranspose2dLayer) Backward(gradOutput *tensor.Tensor, learningRate float32) *tensor.Tensor
  • func (ct *ConvTranspose2dLayer) Forward(inputTensor *tensor.Tensor) *tensor.Tensor
  • func (ct *ConvTranspose2dLayer) GetBias() *tensor.Tensor
  • func (ct *ConvTranspose2dLayer) GetWeights() *tensor.Tensor
  • func (ct *ConvTranspose2dLayer) Parameters() []*tensor.Tensor
  • func (ct *ConvTranspose2dLayer) SetBias(data [][]float32)
  • func (l *ConvTranspose2dLayer) SetBiasAndShape(data []float32, shape []int)
  • func (ct *ConvTranspose2dLayer) SetWeights(data [][]float32)
  • func (l *ConvTranspose2dLayer) SetWeightsAndShape(data []float32, shape []int)
  • func (ct *ConvTranspose2dLayer) ZeroGrad()
• type RMSNorm
  • func NewRMSNorm(features int, eps float32) *RMSNorm
  • func (r *RMSNorm) Backward(gradOutput *tensor.Tensor, learningRate float32) *tensor.Tensor
  • func (r *RMSNorm) Forward(x *tensor.Tensor) *tensor.Tensor
  • func (r *RMSNorm) ForwardMultiThread(inputTensor *tensor.Tensor) *tensor.Tensor
  • func (r *RMSNorm) ForwardSignalThread(inputTensor *tensor.Tensor) *tensor.Tensor
  • func (r *RMSNorm) GetBias() *tensor.Tensor
  • func (r *RMSNorm) GetWeights() *tensor.Tensor
  • func (r *RMSNorm) Parameters() []*tensor.Tensor
  • func (r *RMSNorm) SetBias(data [][]float32)
  • func (r *RMSNorm) SetBiasAndShape(data []float32, shape []int)
  • func (r *RMSNorm) SetWeights(data []float32)
  • func (r *RMSNorm) SetWeightsAndShape(data []float32, shape []int)
  • func (r *RMSNorm) ZeroGrad()
• type TanhLayer
  • func NewTanhLayer() *TanhLayer
  • func (t *TanhLayer) Backward(gradOutput *tensor.Tensor, learningRate float32) *tensor.Tensor
  • func (t *TanhLayer) Forward(input *tensor.Tensor) *tensor.Tensor
  • func (r *TanhLayer) GetBias() *tensor.Tensor
  • func (r *TanhLayer) GetWeights() *tensor.Tensor
  • func (t *TanhLayer) Parameters() []*tensor.Tensor
  • func (t *TanhLayer) SetBias(data [][]float32)
  • func (t *TanhLayer) SetWeights(data [][]float32)
  • func (t *TanhLayer) ZeroGrad()

Types

type ConvTranspose2dLayer

type ConvTranspose2dLayer struct {
	Weights          *tensor.Tensor
	Bias             *tensor.Tensor
	KernelSizeRow    int
	KernelSizeCol    int
	StrideRow        int
	StrideCol        int
	PaddingRow       int
	PaddingCol       int
	OutputPaddingRow int
	OutputPaddingCol int
	InChannels       int
	OutChannels      int
	// contains filtered or unexported fields
}

func NewConvTranspose2dLayer

func NewConvTranspose2dLayer(inChannels, outChannels, kernelSizeRows, kernelSizeCols, strideRows, strideCols, paddingRows, paddingCols, outputPaddingRows, outputPaddingCols int) *ConvTranspose2dLayer

func (*ConvTranspose2dLayer) Backward

func (ct *ConvTranspose2dLayer) Backward(gradOutput *tensor.Tensor, learningRate float32) *tensor.Tensor

func (*ConvTranspose2dLayer) Forward

func (ct *ConvTranspose2dLayer) Forward(inputTensor *tensor.Tensor) *tensor.Tensor

func (*ConvTranspose2dLayer) GetBias

func (ct *ConvTranspose2dLayer) GetBias() *tensor.Tensor

func (*ConvTranspose2dLayer) GetWeights

func (ct *ConvTranspose2dLayer) GetWeights() *tensor.Tensor

func (*ConvTranspose2dLayer) Parameters

func (ct *ConvTranspose2dLayer) Parameters() []*tensor.Tensor

func (*ConvTranspose2dLayer) SetBias

func (ct *ConvTranspose2dLayer) SetBias(data [][]float32)

func (*ConvTranspose2dLayer) SetBiasAndShape

func (l *ConvTranspose2dLayer) SetBiasAndShape(data []float32, shape []int)

func (*ConvTranspose2dLayer) SetWeights

func (ct *ConvTranspose2dLayer) SetWeights(data [][]float32)

func (*ConvTranspose2dLayer) SetWeightsAndShape

func (l *ConvTranspose2dLayer) SetWeightsAndShape(data []float32, shape []int)

func (*ConvTranspose2dLayer) ZeroGrad

func (ct *ConvTranspose2dLayer) ZeroGrad()

type RMSNorm

type RMSNorm struct {
	Weights *tensor.Tensor
	// contains filtered or unexported fields
}

func NewRMSNorm

func NewRMSNorm(features int, eps float32) *RMSNorm

func (*RMSNorm) Backward

func (r *RMSNorm) Backward(gradOutput *tensor.Tensor, learningRate float32) *tensor.Tensor

func (*RMSNorm) Forward

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

func (*RMSNorm) ForwardMultiThread

func (r *RMSNorm) ForwardMultiThread(inputTensor *tensor.Tensor) *tensor.Tensor

func (*RMSNorm) ForwardSignalThread

func (r *RMSNorm) ForwardSignalThread(inputTensor *tensor.Tensor) *tensor.Tensor

func (*RMSNorm) GetBias

func (r *RMSNorm) GetBias() *tensor.Tensor

func (*RMSNorm) GetWeights

func (r *RMSNorm) GetWeights() *tensor.Tensor

func (*RMSNorm) Parameters

func (r *RMSNorm) Parameters() []*tensor.Tensor

func (*RMSNorm) SetBias

func (r *RMSNorm) SetBias(data [][]float32)

func (*RMSNorm) SetBiasAndShape

func (r *RMSNorm) SetBiasAndShape(data []float32, shape []int)

func (*RMSNorm) SetWeights

func (r *RMSNorm) SetWeights(data []float32)

func (*RMSNorm) SetWeightsAndShape

func (r *RMSNorm) SetWeightsAndShape(data []float32, shape []int)

func (*RMSNorm) ZeroGrad

func (r *RMSNorm) ZeroGrad()

type TanhLayer

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

func NewTanhLayer

func NewTanhLayer() *TanhLayer

func (*TanhLayer) Backward

func (t *TanhLayer) Backward(gradOutput *tensor.Tensor, learningRate float32) *tensor.Tensor

func (*TanhLayer) Forward

func (t *TanhLayer) Forward(input *tensor.Tensor) *tensor.Tensor

func (*TanhLayer) GetBias

func (r *TanhLayer) GetBias() *tensor.Tensor

func (*TanhLayer) GetWeights

func (r *TanhLayer) GetWeights() *tensor.Tensor

func (*TanhLayer) Parameters

func (t *TanhLayer) Parameters() []*tensor.Tensor

func (*TanhLayer) SetBias

func (t *TanhLayer) SetBias(data [][]float32)

func (*TanhLayer) SetWeights

func (t *TanhLayer) SetWeights(data [][]float32)

func (*TanhLayer) ZeroGrad

func (t *TanhLayer) ZeroGrad()