Versions in this module Expand all Collapse all v1 v1.1.0 Oct 30, 2023 Changes in this version + type Abs struct + func NewAbs(x O) *Abs[O] + type Add struct + func NewAdd(x1, x2 O) *Add[O] + func (r *Add[O]) Backward(gy mat.Tensor) error + func (r *Add[O]) Forward() (mat.Tensor, error) + func (r *Add[O]) Operands() []mat.Tensor + type AddScalar struct + func NewAddScalar(x1, x2 O) *AddScalar[O] + func (r *AddScalar[O]) Backward(gy mat.Tensor) error + func (r *AddScalar[O]) Forward() (mat.Tensor, error) + func (r *AddScalar[O]) Operands() []mat.Tensor + type Affine struct + func NewAffine(b, w1, x1 O, wxPairs ...O) *Affine[O] + func (a *Affine[O]) Backward(gy mat.Tensor) error + func (a *Affine[O]) Forward() (mat.Tensor, error) + func (a *Affine[O]) Operands() []O + type AppendRows struct + func NewAppendRows(x O, vs ...O) *AppendRows[O] + func (a *AppendRows[O]) Backward(gy mat.Tensor) error + func (a *AppendRows[O]) Forward() (mat.Tensor, error) + func (a *AppendRows[O]) Operands() []O + type At struct + func NewAt(x O, indices ...int) *At[O] + func (r *At[O]) Backward(gy mat.Tensor) error + func (r *At[O]) Forward() (mat.Tensor, error) + func (r *At[O]) Operands() []mat.Tensor + type CELU struct + func NewCELU(x O, alpha O) *CELU[O] + func (r *CELU[O]) Backward(gy mat.Tensor) error + func (r *CELU[O]) Forward() (mat.Tensor, error) + func (r *CELU[O]) Operands() []mat.Tensor + type ColView struct + func NewColView(x O, i int) *ColView[O] + func (r *ColView[O]) Backward(gy mat.Tensor) error + func (r *ColView[O]) Forward() (mat.Tensor, error) + func (r *ColView[O]) Operands() []mat.Tensor + type Concat struct + func NewConcat(xs []O) *Concat[O] + func (r *Concat[O]) Backward(gy mat.Tensor) error + func (r *Concat[O]) Forward() (mat.Tensor, error) + func (r *Concat[O]) Operands() []O + type Copy struct + func NewCopy(x O) *Copy[O] + func (r *Copy[O]) Backward(gy mat.Tensor) error + func (r *Copy[O]) Forward() (mat.Tensor, error) + func (r *Copy[O]) Operands() []mat.Tensor + type Cos struct + func NewCos(x O) *Cos[O] + type Div struct + func NewDiv(x1 O, x2 O) *Div[O] + func (r *Div[O]) Backward(gy mat.Tensor) error + func (r *Div[O]) Forward() (mat.Tensor, error) + func (r *Div[O]) Operands() []mat.Tensor + type DivScalar struct + func NewDivScalar(x1 O, x2 O) *DivScalar[O] + func (r *DivScalar[O]) Backward(gy mat.Tensor) error + func (r *DivScalar[O]) Forward() (mat.Tensor, error) + func (r *DivScalar[O]) Operands() []mat.Tensor + type Dot struct + func NewDot(x1 O, x2 O) *Dot[O] + func (r *Dot[O]) Backward(gy mat.Tensor) error + func (r *Dot[O]) Forward() (mat.Tensor, error) + func (r *Dot[O]) Operands() []mat.Tensor + type Dropout struct + func NewDropout(x O, p float64, randGen *rand.LockedRand) *Dropout[O] + func (r *Dropout[O]) Backward(gy mat.Tensor) error + func (r *Dropout[O]) Forward() (mat.Tensor, error) + func (r *Dropout[O]) Operands() []mat.Tensor + type ELU struct + func NewELU(x O, alpha O) *ELU[O] + func (r *ELU[O]) Backward(gy mat.Tensor) error + func (r *ELU[O]) Forward() (mat.Tensor, error) + func (r *ELU[O]) Operands() []mat.Tensor + type Exp struct + func NewExp(x O) *Exp[O] + func (e *Exp[O]) Backward(gy mat.Tensor) error + func (e *Exp[O]) Forward() (mat.Tensor, error) + func (e *Exp[O]) Operands() []mat.Tensor + type Flatten struct + func NewFlatten(x O) *Flatten[O] + func (r *Flatten[O]) Backward(gy mat.Tensor) error + func (r *Flatten[O]) Forward() (mat.Tensor, error) + func (r *Flatten[O]) Operands() []mat.Tensor + type GELU struct + func NewGELU(x O) *GELU[O] + type HardSigmoid struct + func NewHardSigmoid(x O) *HardSigmoid[O] + type HardTanh struct + func NewHardTanh(x O) *HardTanh[O] + type LeakyReLU struct + func NewLeakyReLU(x, alpha O) *LeakyReLU[O] + func (r *LeakyReLU[O]) Backward(gy mat.Tensor) error + func (r *LeakyReLU[O]) Forward() (mat.Tensor, error) + func (r *LeakyReLU[O]) Operands() []mat.Tensor + type Log struct + func NewLog(x O) *Log[O] + func (l *Log[O]) Backward(gy mat.Tensor) error + func (l *Log[O]) Forward() (mat.Tensor, error) + func (l *Log[O]) Operands() []mat.Tensor + type Max struct + func NewMax(x1 O, x2 O) *Max[O] + func (r *Max[O]) Backward(gy mat.Tensor) error + func (r *Max[O]) Forward() (mat.Tensor, error) + func (r *Max[O]) Operands() []mat.Tensor + type MaxPooling struct + func NewMaxPooling(x O, r, c int) *MaxPooling[O] + func (r *MaxPooling[O]) Backward(gy mat.Tensor) error + func (r *MaxPooling[O]) Forward() (mat.Tensor, error) + func (r *MaxPooling[O]) Operands() []mat.Tensor + type Min struct + func NewMin(x1 O, x2 O) *Min[O] + func (r *Min[O]) Backward(gy mat.Tensor) error + func (r *Min[O]) Forward() (mat.Tensor, error) + func (r *Min[O]) Operands() []mat.Tensor + type Mish struct + func NewMish(x O) *Mish[O] + type Mul struct + func NewMul(x1 O, x2 O) *Mul[O] + func (r *Mul[O]) Backward(gy mat.Tensor) error + func (r *Mul[O]) Forward() (mat.Tensor, error) + func (r *Mul[O]) Operands() []mat.Tensor + type MulT struct + func NewMulT(x1 O, x2 O) *MulT[O] + func (r *MulT[O]) Backward(gy mat.Tensor) error + func (r *MulT[O]) Forward() (mat.Tensor, error) + func (r *MulT[O]) Operands() []mat.Tensor + type Neg struct + func NewNeg(x O) *Neg[O] + type Pow struct + func NewPow(x O, power float64) *Pow[O] + func (r *Pow[O]) Backward(gy mat.Tensor) error + func (r *Pow[O]) Forward() (mat.Tensor, error) + func (r *Pow[O]) Operands() []mat.Tensor + type Prod struct + func NewProd(x1 O, x2 O) *Prod[O] + func (r *Prod[O]) Backward(gy mat.Tensor) error + func (r *Prod[O]) Forward() (mat.Tensor, error) + func (r *Prod[O]) Operands() []mat.Tensor + type ProdScalar struct + func NewProdScalar(x1 O, x2 O) *ProdScalar[O] + func (r *ProdScalar[O]) Backward(gy mat.Tensor) error + func (r *ProdScalar[O]) Forward() (mat.Tensor, error) + func (r *ProdScalar[O]) Operands() []mat.Tensor + type ReLU struct + func NewReLU(x O) *ReLU[O] + type Reciprocal struct + func NewReciprocal(x O) *Reciprocal[O] + type ReduceMax struct + func NewReduceMax(x O) *ReduceMax[O] + func (r *ReduceMax[O]) Backward(gy mat.Tensor) error + func (r *ReduceMax[O]) Forward() (mat.Tensor, error) + func (r *ReduceMax[O]) Operands() []mat.Tensor + type ReduceMean struct + func NewReduceMean(x O) *ReduceMean[O] + func (r *ReduceMean[O]) Backward(gy mat.Tensor) error + func (r *ReduceMean[O]) Forward() (mat.Tensor, error) + func (r *ReduceMean[O]) Operands() []mat.Tensor + type ReduceSum struct + func NewReduceSum(x O) *ReduceSum[O] + func (r *ReduceSum[O]) Backward(gy mat.Tensor) error + func (r *ReduceSum[O]) Forward() (mat.Tensor, error) + func (r *ReduceSum[O]) Operands() []mat.Tensor + type Reshape struct + func NewReshape(x O, r, c int) *Reshape[O] + func (r *Reshape[O]) Backward(gy mat.Tensor) error + func (r *Reshape[O]) Forward() (mat.Tensor, error) + func (r *Reshape[O]) Operands() []mat.Tensor + type ReverseSubScalar struct + func NewReverseSubScalar(x1 O, x2 O) *ReverseSubScalar[O] + func (r *ReverseSubScalar[O]) Backward(gy mat.Tensor) error + func (r *ReverseSubScalar[O]) Forward() (mat.Tensor, error) + func (r *ReverseSubScalar[O]) Operands() []mat.Tensor + type RotateR struct + func NewRotateR(x O, i int) *RotateR[O] + func (r *RotateR[O]) Backward(gy mat.Tensor) error + func (r *RotateR[O]) Forward() (mat.Tensor, error) + func (r *RotateR[O]) Operands() []mat.Tensor + type RowView struct + func NewRowView(x O, i int) *RowView[O] + func (r *RowView[O]) Backward(gy mat.Tensor) error + func (r *RowView[O]) Forward() (mat.Tensor, error) + func (r *RowView[O]) Operands() []mat.Tensor + type SELU struct + func NewSELU(x O, alpha, scale O) *SELU[O] + func (r *SELU[O]) Backward(gy mat.Tensor) error + func (r *SELU[O]) Forward() (mat.Tensor, error) + func (r *SELU[O]) Operands() []mat.Tensor + type ScalarMax struct + func NewScalarMax(xs []O) *ScalarMax[O] + func (r *ScalarMax[O]) Backward(gy mat.Tensor) error + func (r *ScalarMax[O]) Forward() (mat.Tensor, error) + func (r *ScalarMax[O]) Operands() []O + type Sigmoid struct + func NewSigmoid(x O) *Sigmoid[O] + func (l *Sigmoid[O]) Backward(gy mat.Tensor) error + func (l *Sigmoid[O]) Forward() (mat.Tensor, error) + func (l *Sigmoid[O]) Operands() []mat.Tensor + type Sin struct + func NewSin(x O) *Sin[O] + type Slice struct + func NewSlice(x O, fromRow, fromCol, toRow, toCol int) *Slice[O] + func (s *Slice[O]) Backward(gy mat.Tensor) error + func (s *Slice[O]) Forward() (mat.Tensor, error) + func (s *Slice[O]) Operands() []mat.Tensor + type SoftPlus struct + func NewSoftPlus(x O, beta, threshold O) *SoftPlus[O] + func (r *SoftPlus[O]) Backward(gy mat.Tensor) error + func (r *SoftPlus[O]) Forward() (mat.Tensor, error) + func (r *SoftPlus[O]) Operands() []mat.Tensor + type SoftShrink struct + func NewSoftShrink(x O, lambda O) *SoftShrink[O] + func (r *SoftShrink[O]) Backward(gy mat.Tensor) error + func (r *SoftShrink[O]) Forward() (mat.Tensor, error) + func (r *SoftShrink[O]) Operands() []mat.Tensor + type Softmax struct + func NewSoftmax(x O) *Softmax[O] + func (r *Softmax[O]) Backward(gy mat.Tensor) error + func (r *Softmax[O]) Forward() (mat.Tensor, error) + func (r *Softmax[O]) Operands() []mat.Tensor + type Softsign struct + func NewSoftsign(x O) *Softsign[O] + type SparseMax struct + func NewSparseMax(x O) *SparseMax[O] + func (r *SparseMax[O]) Backward(gy mat.Tensor) error + func (r *SparseMax[O]) Forward() (mat.Tensor, error) + func (r *SparseMax[O]) Operands() []mat.Tensor + type SparseMaxLoss struct + func NewSparseMaxLoss(x O) *SparseMaxLoss[O] + func (r *SparseMaxLoss[O]) Backward(gy mat.Tensor) error + func (r *SparseMaxLoss[O]) Forward() (mat.Tensor, error) + func (r *SparseMaxLoss[O]) Operands() []mat.Tensor + type Sqrt struct + func NewSqrt(x O) *Sqrt[O] + func (r *Sqrt[O]) Backward(gy mat.Tensor) error + func (r *Sqrt[O]) Forward() (mat.Tensor, error) + func (r *Sqrt[O]) Operands() []mat.Tensor + type Square struct + func NewSquare(x O) *Square[O] + type Stack struct + func NewStack(xs []O) *Stack[O] + func (r *Stack[O]) Backward(gy mat.Tensor) error + func (r *Stack[O]) Forward() (mat.Tensor, error) + func (r *Stack[O]) Operands() []O + type Sub struct + func NewSub(x1 O, x2 O) *Sub[O] + func (r *Sub[O]) Backward(gy mat.Tensor) error + func (r *Sub[O]) Forward() (mat.Tensor, error) + func (r *Sub[O]) Operands() []mat.Tensor + type SubScalar struct + func NewSubScalar(x1 O, x2 O) *SubScalar[O] + func (r *SubScalar[O]) Backward(gy mat.Tensor) error + func (r *SubScalar[O]) Forward() (mat.Tensor, error) + func (r *SubScalar[O]) Operands() []mat.Tensor + type Swish struct + func NewSiLU(x O) *Swish[O] + func NewSwish(x O) *Swish[O] + func (l *Swish[O]) Backward(gy mat.Tensor) error + func (l *Swish[O]) Forward() (mat.Tensor, error) + func (l *Swish[O]) Operands() []mat.Tensor + type SwishB struct + func NewSwishB(x O, beta O) *SwishB[O] + func (r *SwishB[O]) Backward(gy mat.Tensor) error + func (r *SwishB[O]) Forward() (mat.Tensor, error) + func (r *SwishB[O]) Operands() []mat.Tensor + type Tan struct + func NewTan(x O) *Tan[O] + type Tanh struct + func NewTanh(x O) *Tanh[O] + type Threshold struct + func NewThreshold(x O, threshold, k O) *Threshold[O] + func (r *Threshold[O]) Backward(gy mat.Tensor) error + func (r *Threshold[O]) Forward() (mat.Tensor, error) + func (r *Threshold[O]) Operands() []mat.Tensor + type Transpose struct + func NewTranspose(x O) *Transpose[O] + func (r *Transpose[O]) Backward(gy mat.Tensor) error + func (r *Transpose[O]) Forward() (mat.Tensor, error) + func (r *Transpose[O]) Operands() []mat.Tensor + type UnaryElementwise struct + func (r *UnaryElementwise[O]) Backward(gy mat.Tensor) error + func (r *UnaryElementwise[O]) Forward() (mat.Tensor, error) + func (r *UnaryElementwise[O]) Operands() []mat.Tensor