layers

package

v0.0.0-...-d606bfb Latest Latest Go to latest Published: Dec 19, 2021 License: MIT Imports: 2 Imported by: 0

Details

Valid go.mod file

The Go module system was introduced in Go 1.11 and is the official dependency management solution for Go.
Redistributable license

Redistributable licenses place minimal restrictions on how software can be used, modified, and redistributed.
Tagged version

Modules with tagged versions give importers more predictable builds.
Stable version

When a project reaches major version v1 it is considered stable.
Learn more about best practices

This section is empty.

This section is empty.

This section is empty.

type Conv2D struct {
	KernelSize struct {
		X int
		Y int
	}
	Depth  int
	Stride int
	// contains filtered or unexported fields
}

func (e *Conv2D) Backward(error [][][]float64, lr float64) [][][]float64

func (e *Conv2D) Forward(in [][][]float64) (out [][][]float64)

type FC struct {
	Out int
	// contains filtered or unexported fields
}

Fully Connected layer

func (e FC) Backward(err [][][]float64, lr float64) [][][]float64

func (e *FC) Forward(in [][][]float64) (out [][][]float64)

FC Forward function

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

Flatten layer

func (e *Flatten) Backward(err [][][]float64, float642 float64) [][][]float64

Flatten backward function

func (e *Flatten) Forward(in [][][]float64) [][][]float64

Flattern layer forward function

type TestConv struct {
	KernelSize struct {
		X int
		Y int
	}
	Depth  int
	Stride int
	// contains filtered or unexported fields
}

func (e *TestConv) Forward(in [][][]float64) (out [][][]float64)