LearnedPositionalEmbedding class

Inheritance

Constructors

LearnedPositionalEmbedding(int maxLen, int embedDim, {Device device = Device.CPU, int seed = 0})

Properties

embedDim int
final
hashCode int
The hash code for this object.
no setterinherited
maxLen int
final
runtimeType Type
A representation of the runtime type of the object.
no setterinherited
table Embedding
final
training bool
Whether this module is in training mode. Layers that behave differently between training and inference (e.g. Dropout) read this flag in their call method. Defaults to training mode.
getter/setter pairinherited

Methods

call(Tensor x, {int startPos = 0}) Tensor
eval() → void
Put this module (and any registered submodules) into evaluation mode.
inherited
noSuchMethod(Invocation invocation) → dynamic
Invoked when a nonexistent method or property is accessed.
inherited
parameters() List<Tensor>
Trainable tensors owned by this module (and its submodules).
override
submodules() List<Module>
Submodules owned by this module. Subclasses that compose other modules should override this so train() / eval() propagate. Default: empty.
override
toString() String
A string representation of this object.
inherited
train() → void
Put this module (and any registered submodules) into training mode.
inherited
zeroGrad() → void
Zero every parameter's gradient. Safe to call before each backward.
inherited

Operators

operator ==(Object other) bool
The equality operator.
inherited