layer

package

v1.0.0 Latest Latest Go to latest Published: Jun 15, 2023 License: MIT Imports: 5 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 ¶

type Dense
- func NewDense(output int) *Dense
- func (layer *Dense) Args() map[string]float32
- func (b Dense) Class() string
- func (layer *Dense) Forward(x *tensor.Tensor) *tensor.Tensor
- func (b Dense) Name() string
- func (layer *Dense) Params() map[string]*tensor.Tensor
- func (b Dense) SetName(name string)
type Dropout
- func NewDropout(keep float64) *Dropout
- func (layer *Dropout) Args() map[string]float32
- func (b Dropout) Class() string
- func (layer *Dropout) Forward(x *tensor.Tensor, train bool) *tensor.Tensor
- func (b Dropout) Name() string
- func (b Dropout) Params() map[string]*tensor.Tensor
- func (b Dropout) SetName(name string)
type Flatten
- func NewFlatten() *Flatten
- func (b Flatten) Args() map[string]float32
- func (b Flatten) Class() string
- func (layer *Flatten) Forward(x *tensor.Tensor) *tensor.Tensor
- func (b Flatten) Name() string
- func (b Flatten) Params() map[string]*tensor.Tensor
- func (b Flatten) SetName(name string)
type Layer
- func LoadDense(name string, params map[string]*pb.Dense, args map[string]float32) Layer
- func LoadDropout(name string, _ map[string]*pb.Dense, args map[string]float32) Layer
- func LoadFlatten(name string, params map[string]*pb.Dense, args map[string]float32) Layer
- func LoadLstm(name string, params map[string]*pb.Dense, args map[string]float32) Layer
- func LoadNor(name string, params map[string]*pb.Dense, args map[string]float32) Layer
- func LoadRnn(name string, params map[string]*pb.Dense, args map[string]float32) Layer
- func LoadSelfAttention(name string, params map[string]*pb.Dense, args map[string]float32) Layer
type Lstm
- func NewLstm(featureSize, steps, hidden int) *Lstm
- func (layer *Lstm) Args() map[string]float32
- func (b Lstm) Class() string
- func (layer *Lstm) Forward(x, h, c *tensor.Tensor) (*tensor.Tensor, *tensor.Tensor, *tensor.Tensor)
- func (b Lstm) Name() string
- func (layer *Lstm) Params() map[string]*tensor.Tensor
- func (b Lstm) SetName(name string)
type Nor
- func NewNor() *Nor
- func (b Nor) Args() map[string]float32
- func (b Nor) Class() string
- func (layer *Nor) Forward(x *tensor.Tensor) *tensor.Tensor
- func (b Nor) Name() string
- func (b Nor) Params() map[string]*tensor.Tensor
- func (b Nor) SetName(name string)
type Rnn
- func NewRnn(featureSize, steps, hidden int) *Rnn
- func (layer *Rnn) Args() map[string]float32
- func (b Rnn) Class() string
- func (layer *Rnn) Forward(x, h *tensor.Tensor) (*tensor.Tensor, *tensor.Tensor)
- func (b Rnn) Name() string
- func (layer *Rnn) Params() map[string]*tensor.Tensor
- func (b Rnn) SetName(name string)
type SelfAttention
- func NewSelfAttention(steps, dims, heads int) *SelfAttention
- func (layer *SelfAttention) Args() map[string]float32
- func (b SelfAttention) Class() string
- func (layer *SelfAttention) Forward(q, k *tensor.Tensor) *tensor.Tensor
- func (b SelfAttention) Name() string
- func (layer *SelfAttention) Params() map[string]*tensor.Tensor
- func (b SelfAttention) SetName(name string)

Constants ¶

This section is empty.

Variables ¶

This section is empty.

Functions ¶

This section is empty.

Types ¶

type Dense ¶

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

func NewDense ¶

func NewDense(output int) *Dense

func (*Dense) Args ¶

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

func (Dense) Class ¶

func (b Dense) Class() string

func (*Dense) Forward ¶

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

func (Dense) Name ¶

func (b Dense) Name() string

func (*Dense) Params ¶

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

func (Dense) SetName ¶

func (b Dense) SetName(name string)

type Dropout ¶

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

func NewDropout ¶

func NewDropout(keep float64) *Dropout

func (*Dropout) Args ¶

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

func (Dropout) Class ¶

func (b Dropout) Class() string

func (*Dropout) Forward ¶

func (layer *Dropout) Forward(x *tensor.Tensor, train bool) *tensor.Tensor

func (Dropout) Name ¶

func (b Dropout) Name() string

func (Dropout) Params ¶

func (b Dropout) Params() map[string]*tensor.Tensor

func (Dropout) SetName ¶

func (b Dropout) SetName(name string)

type Flatten ¶ added in v1.0.0

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

func NewFlatten ¶ added in v1.0.0

func NewFlatten() *Flatten

func (Flatten) Args ¶ added in v1.0.0

func (b Flatten) Args() map[string]float32

func (Flatten) Class ¶ added in v1.0.0

func (b Flatten) Class() string

func (*Flatten) Forward ¶ added in v1.0.0

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

func (Flatten) Name ¶ added in v1.0.0

func (b Flatten) Name() string

func (Flatten) Params ¶ added in v1.0.0

func (b Flatten) Params() map[string]*tensor.Tensor

func (Flatten) SetName ¶ added in v1.0.0

func (b Flatten) SetName(name string)

type Layer ¶

type Layer interface {
	Params() map[string]*tensor.Tensor
	Class() string
	SetName(name string)
	Name() string
	Args() map[string]float32
}

func LoadDense ¶

func LoadDense(name string, params map[string]*pb.Dense, args map[string]float32) Layer

func LoadDropout ¶

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

func LoadFlatten ¶ added in v1.0.0

func LoadFlatten(name string, params map[string]*pb.Dense, args map[string]float32) Layer

func LoadLstm ¶ added in v1.0.0

func LoadLstm(name string, params map[string]*pb.Dense, args map[string]float32) Layer

func LoadNor ¶ added in v1.0.0

func LoadNor(name string, params map[string]*pb.Dense, args map[string]float32) Layer

func LoadRnn ¶

func LoadRnn(name string, params map[string]*pb.Dense, args map[string]float32) Layer

func LoadSelfAttention ¶ added in v1.0.0

func LoadSelfAttention(name string, params map[string]*pb.Dense, args map[string]float32) Layer

type Lstm ¶ added in v1.0.0

type Lstm struct {
	Wi, Bi *tensor.Tensor
	Wf, Bf *tensor.Tensor
	Wg, Bg *tensor.Tensor
	Wo, Bo *tensor.Tensor
	// contains filtered or unexported fields
}

func NewLstm ¶ added in v1.0.0

func NewLstm(featureSize, steps, hidden int) *Lstm

func (*Lstm) Args ¶ added in v1.0.0

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

func (Lstm) Class ¶ added in v1.0.0

func (b Lstm) Class() string

func (*Lstm) Forward ¶ added in v1.0.0

func (layer *Lstm) Forward(x, h, c *tensor.Tensor) (*tensor.Tensor, *tensor.Tensor, *tensor.Tensor)

func (Lstm) Name ¶ added in v1.0.0

func (b Lstm) Name() string

func (*Lstm) Params ¶ added in v1.0.0

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

func (Lstm) SetName ¶ added in v1.0.0

func (b Lstm) SetName(name string)

type Nor ¶ added in v1.0.0

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

func NewNor ¶ added in v1.0.0

func NewNor() *Nor

func (Nor) Args ¶ added in v1.0.0

func (b Nor) Args() map[string]float32

func (Nor) Class ¶ added in v1.0.0

func (b Nor) Class() string

func (*Nor) Forward ¶ added in v1.0.0

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

func (Nor) Name ¶ added in v1.0.0

func (b Nor) Name() string

func (Nor) Params ¶ added in v1.0.0

func (b Nor) Params() map[string]*tensor.Tensor

func (Nor) SetName ¶ added in v1.0.0

func (b Nor) SetName(name string)

type Rnn ¶

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

func NewRnn ¶

func NewRnn(featureSize, steps, hidden int) *Rnn

func (*Rnn) Args ¶

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

func (Rnn) Class ¶

func (b Rnn) Class() string

func (*Rnn) Forward ¶

func (layer *Rnn) Forward(x, h *tensor.Tensor) (*tensor.Tensor, *tensor.Tensor)

func (Rnn) Name ¶

func (b Rnn) Name() string

func (*Rnn) Params ¶

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

func (Rnn) SetName ¶

func (b Rnn) SetName(name string)

type SelfAttention ¶ added in v1.0.0

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

func NewSelfAttention ¶ added in v1.0.0

func NewSelfAttention(steps, dims, heads int) *SelfAttention

func (*SelfAttention) Args ¶ added in v1.0.0

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

func (SelfAttention) Class ¶ added in v1.0.0

func (b SelfAttention) Class() string

func (*SelfAttention) Forward ¶ added in v1.0.0

func (layer *SelfAttention) Forward(q, k *tensor.Tensor) *tensor.Tensor

func (SelfAttention) Name ¶ added in v1.0.0

func (b SelfAttention) Name() string

func (*SelfAttention) Params ¶ added in v1.0.0

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

func (SelfAttention) SetName ¶ added in v1.0.0

func (b SelfAttention) SetName(name string)

Source Files ¶

View all Source files

Directories ¶

Path	Synopsis
activation

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