pooling

Documentation

Index

• type MaxPooling
  • func NewMax[T mat.DType](rows, columns int) *MaxPooling[T]
  • func (m *MaxPooling[T]) Forward(xs ...ag.Node[T]) []ag.Node[T]

Types

type MaxPooling

type MaxPooling[T mat.DType] struct {
	nn.BaseModel
	Rows    int
	Columns int
}

MaxPooling is a parameter-free model used to instantiate a new Processor.

func NewMax

func NewMax[T mat.DType](rows, columns int) *MaxPooling[T]

NewMax returns a new model.

func (*MaxPooling[T]) Forward

func (m *MaxPooling[T]) Forward(xs ...ag.Node[T]) []ag.Node[T]

Forward performs the forward step for each input node and returns the result. The max pooling is applied independently to each input.