pooling

Documentation

Index

• type MaxPooling
  • func NewMax(rows, columns int) *MaxPooling
  • func (m *MaxPooling) Forward(xs ...mat.Tensor) []mat.Tensor

Types

type MaxPooling

type MaxPooling struct {
	nn.Module
	Rows    int
	Columns int
}

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

func NewMax

func NewMax(rows, columns int) *MaxPooling

NewMax returns a new model.

func (*MaxPooling) Forward

func (m *MaxPooling) Forward(xs ...mat.Tensor) []mat.Tensor

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