net

package

v1.0.3 Latest Latest Go to latest Published: May 8, 2025 License: MIT Imports: 18 Imported by: 0

Details

This section is empty.

This section is empty.

func RegisterLoadFunc(class string, fn loadFunc)

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

func New(device consts.DeviceType) *Net

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

func (n *Net) Clear()

func (n *Net) GetOptimizer() optimizer.Optimizer

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

func (n *Net) Load(dir string) error

func (n *Net) ParamCount() uint64

func (n *Net) Params() []*tensor.Tensor

func (n *Net) ReadFrom(r io.ReaderAt, size int64) (int64, error)

func (n *Net) Save(dir string) error

func (n *Net) SetDevice(device consts.DeviceType)

func (n *Net) SetOptimizer(optm optimizer.Optimizer)

func (n *Net) ToScalarType(t consts.ScalarType)

func (n *Net) WriteTo(w io.Writer) (int64, error)