xloss

Documentation

Index

• type Ops
  • func Stage(handle *xtra.Handle, mode xtra.XLossMode, managed bool) (*Ops, error)
  • func (o *Ops) Error(handle *xtra.Handle, dx, y, dy *tensor.Volume, alpha, beta float64) error
  • func (o *Ops) Loss() float32

Types

type Ops

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

Ops contains the operation to do the loss function

func Stage

func Stage(handle *xtra.Handle, mode xtra.XLossMode, managed bool) (*Ops, error)

Stage stages the loss operation

func (*Ops) Error

func (o *Ops) Error(handle *xtra.Handle, dx, y, dy *tensor.Volume, alpha, beta float64) error

Error calculates the MSE error dx will get the errors y is the target values dy is the network output

func (*Ops) Loss

func (o *Ops) Loss() float32

Loss returns the loss function found. Should be called after the error is found