dropout

Documentation

Index

• type Ops
  • func Stage(handle *cudnn.Handler, x *tensor.Volume, dropout float32, seed uint64) (*Ops, error)
  • func (op *Ops) BackProp(handle *cudnn.Handler, dx, dy *tensor.Volume) error
  • func (op *Ops) ForwardProp(handle *cudnn.Handler, x, y *tensor.Volume) error

Types

type Ops

type Ops struct {
	// contains filtered or unexported fields
}

Ops does the operation. It also holds the memory to run the tensor op

func Stage

func Stage(handle *cudnn.Handler, x *tensor.Volume, dropout float32, seed uint64) (*Ops, error)

Stage stages the op

func (*Ops) BackProp

func (op *Ops) BackProp(handle *cudnn.Handler, dx, dy *tensor.Volume) error

BackProp does the back propagation for dropoutlayer

func (*Ops) ForwardProp

func (op *Ops) ForwardProp(handle *cudnn.Handler, x, y *tensor.Volume) error

ForwardProp does the feed forward