transform

Documentation

Index

• type Ops
  • func Stage(nbDims uint32, destfrmt gocudnn.TensorFormat, padbefore, padafter []int32, ...) (op *Ops, err error)
  • func (t *Ops) Transform(handle cudnn.Handler, alpha float64, src *tensor.Volume, beta float64, ...) error

Types

type Ops

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

Ops is the struct that holds the descriptor for the transform operation

func Stage

func Stage(nbDims uint32, destfrmt gocudnn.TensorFormat, padbefore, padafter []int32, fold []uint32, direction gocudnn.FoldingDirection) (op *Ops, err error)

Stage stages transform operation

func (*Ops) Transform

func (t *Ops) Transform(handle cudnn.Handler, alpha float64, src *tensor.Volume, beta float64, dest *tensor.Volume) error

Transform does the transform operation