convolution

Documentation

Index

• type Config
• type Model
  • func New(config Config) *Model
  • func (m *Model) NewProc(ctx nn.Context) nn.Processor
• type Processor
  • func (p *Processor) Forward(xs ...ag.Node) []ag.Node
  • func (p *Processor) SetConcurrentComputations(value bool)

Types

type Config

type Config struct {
	KernelSizeX    int
	KernelSizeY    int
	XStride        int
	YStride        int
	InputChannels  int
	OutputChannels int
	Mask           []int
	Activation     ag.OpName
}

type Model

type Model struct {
	Config
	K []*nn.Param `type:"weights"`
	B []*nn.Param `type:"biases"`
}

func New

func New(config Config) *Model

func (*Model) NewProc

func (m *Model) NewProc(ctx nn.Context) nn.Processor

type Processor

type Processor struct {
	nn.BaseProcessor
	Config
	// contains filtered or unexported fields
}

func (*Processor) Forward

func (p *Processor) Forward(xs ...ag.Node) []ag.Node

func (*Processor) SetConcurrentComputations

func (p *Processor) SetConcurrentComputations(value bool)