linear

package

v1.1.0 Latest Latest Go to latest Published: Oct 30, 2023 License: BSD-2-Clause Imports: 5 Imported by: 9

Details

This section is empty.

This section is empty.

This section is empty.

type Model struct {
	nn.Module
	W *nn.Param
	B *nn.Param
}

Model contains the serializable parameters.

func New[T float.DType](in, out int) *Model

New returns a new model with parameters initialized to zeros.

func (m *Model) Forward(xs ...mat.Tensor) []mat.Tensor

Forward performs the forward step for each input node and returns the result.

func (m *Model) WithBiasGrad(value bool) *Model

WithBiasGrad allows you to enable or disable gradient propagation on bias (enabled by default).