Documentation
¶
Index ¶
- func Load(class string) func(string, map[string]*pb.Dense, map[string]*pb.Dense) layer.Layer
- type Activation
- type ReLU
- func (ReLU) Args() map[string]mat.Matrix
- func (layer ReLU) Backward(context []mat.Matrix, grad mat.Matrix) (valueGrad mat.Matrix, paramsGrad *params.Params)
- func (layer ReLU) Class() string
- func (ReLU) Context() *params.Params
- func (layer ReLU) Forward(input mat.Matrix, _ bool) (context []mat.Matrix, output mat.Matrix)
- func (layer ReLU) Name() string
- func (ReLU) Params() *params.Params
- func (layer ReLU) Print()
- func (layer ReLU) SetName(name string)
- func (ReLU) SetTraining(bool)
- type Sigmoid
- func (Sigmoid) Args() map[string]mat.Matrix
- func (layer Sigmoid) Backward(context []mat.Matrix, grad mat.Matrix) (valueGrad mat.Matrix, paramsGrad *params.Params)
- func (layer Sigmoid) Class() string
- func (Sigmoid) Context() *params.Params
- func (layer Sigmoid) Forward(input mat.Matrix, _ bool) (context []mat.Matrix, output mat.Matrix)
- func (layer Sigmoid) Name() string
- func (Sigmoid) Params() *params.Params
- func (layer Sigmoid) Print()
- func (layer Sigmoid) SetName(name string)
- func (Sigmoid) SetTraining(bool)
- type Softplus
- func (Softplus) Args() map[string]mat.Matrix
- func (layer Softplus) Backward(context []mat.Matrix, grad mat.Matrix) (valueGrad mat.Matrix, paramsGrad *params.Params)
- func (layer Softplus) Class() string
- func (Softplus) Context() *params.Params
- func (layer Softplus) Forward(input mat.Matrix, _ bool) (context []mat.Matrix, output mat.Matrix)
- func (layer Softplus) Name() string
- func (Softplus) Params() *params.Params
- func (layer Softplus) Print()
- func (layer Softplus) SetName(name string)
- func (Softplus) SetTraining(bool)
- type Tanh
- func (Tanh) Args() map[string]mat.Matrix
- func (layer Tanh) Backward(context []mat.Matrix, grad mat.Matrix) (valueGrad mat.Matrix, paramsGrad *params.Params)
- func (layer Tanh) Class() string
- func (Tanh) Context() *params.Params
- func (layer Tanh) Forward(input mat.Matrix, _ bool) (context []mat.Matrix, output mat.Matrix)
- func (layer Tanh) Name() string
- func (Tanh) Params() *params.Params
- func (layer Tanh) Print()
- func (layer Tanh) SetName(name string)
- func (Tanh) SetTraining(bool)
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
Types ¶
type Activation ¶
func NewReLU ¶
func NewReLU() Activation
func NewSigmoid ¶
func NewSigmoid() Activation
func NewSoftplus ¶
func NewSoftplus() Activation
func NewTanh ¶
func NewTanh() Activation
type ReLU ¶
type ReLU struct {
// contains filtered or unexported fields
}
func (ReLU) SetTraining ¶
func (ReLU) SetTraining(bool)
type Sigmoid ¶
type Sigmoid struct {
// contains filtered or unexported fields
}
func (Sigmoid) SetTraining ¶
func (Sigmoid) SetTraining(bool)
type Softplus ¶
type Softplus struct {
// contains filtered or unexported fields
}
func (Softplus) SetTraining ¶
func (Softplus) SetTraining(bool)
Source Files
Click to show internal directories.
Click to hide internal directories.