convolution2d

type Config struct {
	KernelSizeX    int
	KernelSizeY    int
	XStride        int
	YStride        int
	InputChannels  int
	OutputChannels int
	Mask           []int
	DepthWise      bool // Special case od depthwise convolution, where outputchannels == inputchannels
	Activation     activation.Activation
}

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