activation

package

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

Details

Valid go.mod file
Redistributable license
Tagged version
Stable version
Learn more about best practices

Repository

github.com/lwch/tnn

Links

Open Source Insights

Documentation ¶

Index ¶

func LoadGelu(name string, _ []*tensor.Tensor, args map[string]float32) layer.Layer
func LoadRelu(name string, _ []*tensor.Tensor, _ map[string]float32) layer.Layer
func LoadSigmoid(name string, _ []*tensor.Tensor, _ map[string]float32) layer.Layer
func LoadTanh(name string, _ []*tensor.Tensor, _ map[string]float32) layer.Layer
type GeLU
- func NewGeLU(tanh bool) *GeLU
type ReLU
- func NewReLU() *ReLU
type Sigmoid
- func NewSigmoid() *Sigmoid
type Tanh
- func NewTanh() *Tanh

Constants ¶

This section is empty.

Variables ¶

This section is empty.

Functions ¶

func LoadGelu ¶ added in v1.0.1

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

func LoadRelu ¶ added in v1.0.0

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

func LoadSigmoid ¶ added in v1.0.0

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

func LoadTanh ¶ added in v1.0.2

func LoadTanh(name string, _ []*tensor.Tensor, _ map[string]float32) layer.Layer

Types ¶

type GeLU ¶ added in v1.0.1

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

func NewGeLU ¶ added in v1.0.1

func NewGeLU(tanh bool) *GeLU

func (*GeLU) Args ¶ added in v1.0.1

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

func (GeLU) Class ¶ added in v1.0.1

func (layer GeLU) Class() string

func (*GeLU) Clone ¶ added in v1.0.2

func (layer *GeLU) Clone() layer.Layer

func (*GeLU) Forward ¶ added in v1.0.1

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

func (GeLU) Freeze ¶ added in v1.0.2

func (GeLU) Freeze()

func (GeLU) Name ¶ added in v1.0.1

func (layer GeLU) Name() string

func (GeLU) Params ¶ added in v1.0.1

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

func (GeLU) Reset ¶ added in v1.0.2

func (GeLU) Reset()

func (GeLU) SetName ¶ added in v1.0.1

func (layer GeLU) SetName(name string)

func (GeLU) ToScalarType ¶ added in v1.0.2

func (GeLU) ToScalarType(t consts.ScalarType)

func (GeLU) Unfreeze ¶ added in v1.0.2

func (GeLU) Unfreeze()

type ReLU ¶

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

func NewReLU ¶

func NewReLU() *ReLU

func (ReLU) Args ¶

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

func (ReLU) Class ¶

func (layer ReLU) Class() string

func (*ReLU) Clone ¶ added in v1.0.2

func (layer *ReLU) Clone() layer.Layer

func (*ReLU) Forward ¶

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

func (ReLU) Freeze ¶ added in v1.0.2

func (ReLU) Freeze()

func (ReLU) Name ¶

func (layer ReLU) Name() string

func (ReLU) Params ¶

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

func (ReLU) Reset ¶ added in v1.0.2

func (ReLU) Reset()

func (ReLU) SetName ¶

func (layer ReLU) SetName(name string)

func (ReLU) ToScalarType ¶ added in v1.0.2

func (ReLU) ToScalarType(t consts.ScalarType)

func (ReLU) Unfreeze ¶ added in v1.0.2

func (ReLU) Unfreeze()

type Sigmoid ¶

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

func NewSigmoid ¶

func NewSigmoid() *Sigmoid

func (Sigmoid) Args ¶

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

func (Sigmoid) Class ¶

func (layer Sigmoid) Class() string

func (*Sigmoid) Clone ¶ added in v1.0.2

func (layer *Sigmoid) Clone() layer.Layer

func (*Sigmoid) Forward ¶

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

func (Sigmoid) Freeze ¶ added in v1.0.2

func (Sigmoid) Freeze()

func (Sigmoid) Name ¶

func (layer Sigmoid) Name() string

func (Sigmoid) Params ¶

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

func (Sigmoid) Reset ¶ added in v1.0.2

func (Sigmoid) Reset()

func (Sigmoid) SetName ¶

func (layer Sigmoid) SetName(name string)

func (Sigmoid) ToScalarType ¶ added in v1.0.2

func (Sigmoid) ToScalarType(t consts.ScalarType)

func (Sigmoid) Unfreeze ¶ added in v1.0.2

func (Sigmoid) Unfreeze()

type Tanh ¶

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

func NewTanh ¶

func NewTanh() *Tanh

func (Tanh) Args ¶

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

func (Tanh) Class ¶

func (layer Tanh) Class() string

func (*Tanh) Clone ¶ added in v1.0.2

func (layer *Tanh) Clone() layer.Layer

func (*Tanh) Forward ¶

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

func (Tanh) Freeze ¶ added in v1.0.2

func (Tanh) Freeze()

func (Tanh) Name ¶

func (layer Tanh) Name() string

func (Tanh) Params ¶

func (layer Tanh) Params() []*tensor.Tensor

func (Tanh) Reset ¶ added in v1.0.2

func (Tanh) Reset()

func (Tanh) SetName ¶

func (layer Tanh) SetName(name string)

func (Tanh) ToScalarType ¶ added in v1.0.2

func (Tanh) ToScalarType(t consts.ScalarType)

func (Tanh) Unfreeze ¶ added in v1.0.2

func (Tanh) Unfreeze()

Source Files ¶

View all Source files

?	: This menu
/	: Search site
f or F	: Jump to
y or Y	: Canonical URL