activation

package

v1.0.1 Latest Latest Go to latest Published: Jun 17, 2023 License: MIT Imports: 3 Imported by: 0

Details

This section is empty.

This section is empty.

func LoadGelu(name string, _ map[string]*pb.Dense, args map[string]float32) layer.Layer

func LoadRelu(name string, _ map[string]*pb.Dense, _ map[string]float32) layer.Layer

func LoadSigmoid(name string, _ map[string]*pb.Dense, _ map[string]float32) layer.Layer

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

func NewGeLU(tanh bool) *GeLU

func (layer *GeLU) Args() map[string]float32

func (layer GeLU) Class() string

func (layer *GeLU) Forward(x *tensor.Tensor) *tensor.Tensor

func (layer GeLU) Name() string

func (layer GeLU) Params() map[string]*tensor.Tensor

func (layer GeLU) SetName(name string)

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

func NewReLU() *ReLU

func (ReLU) Args() map[string]float32

func (layer ReLU) Class() string

func (layer *ReLU) Forward(x *tensor.Tensor) *tensor.Tensor

func (layer ReLU) Name() string

func (layer ReLU) Params() map[string]*tensor.Tensor

func (layer ReLU) SetName(name string)

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

func NewSigmoid() *Sigmoid

func (Sigmoid) Args() map[string]float32

func (layer Sigmoid) Class() string

func (layer *Sigmoid) Forward(x *tensor.Tensor) *tensor.Tensor

func (layer Sigmoid) Name() string

func (layer Sigmoid) Params() map[string]*tensor.Tensor

func (layer Sigmoid) SetName(name string)