net

package

v0.0.1 Latest Latest Go to latest Published: May 9, 2023 License: MIT Imports: 6 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

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

func New() *Net

func (n *Net) Add(layer layer.Layer)

func (n *Net) Backward(grad mat.Matrix, ctx [][]mat.Matrix) []*params.Params

func (n *Net) Forward(input mat.Matrix, isTraining bool) (ret mat.Matrix, context [][]mat.Matrix)

func (n *Net) Layers() []layer.Layer

func (n *Net) LoadLayers(layers []*pb.Layer)

func (n *Net) ParamCount() uint64

func (n *Net) Params() []*params.Params

func (n *Net) Print()

func (n *Net) SaveLayers() []*pb.Layer

func (n *Net) Set(layer ...layer.Layer)