convolution1d

Documentation

Index

• type Config
• type Model
  • func New[T float.DType](config Config) *Model
  • func (m *Model) Forward(xs ...mat.Tensor) []mat.Tensor

Types

type Config

type Config struct {
	KernelSizeX    int
	KernelSizeY    int
	YStride        int
	InputChannels  int
	OutputChannels int
	Mask           []int
	DepthWise      bool // Special case od depth-wise convolution, where output channels == input channels
	Activation     activation.Activation
}

Config provides configuration settings for a convolution Model.

type Model

type Model struct {
	nn.Module
	Config Config
	K      []*nn.Param
	B      []*nn.Param
}

Model contains the serializable parameters for a convolutional neural network model.

func New

func New[T float.DType](config Config) *Model

New returns a new convolution Model, initialized according to the given configuration.

func (*Model) Forward

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

Forward performs the forward step for each input node and returns the result.