Embedding

Emebedding ID into vector

struct Embedding (

T

alias Storage

) if (

isFloatingPoint!T

) {

import mir.ndslice : slice;;

Variable!(T, 2, Storage) weight;

this(uint nVocab, uint nEmbed);

void resetParameters();

auto opCall(Variable!(int, 1, Storage) ids);

}

Constructors

this this(uint nVocab, uint nEmbed): Undocumented in source.

Members

Functions

opCall auto opCall(Variable!(int, 1, Storage) ids): Undocumented in source. Be warned that the author may not have intended to support it.
resetParameters void resetParameters()
nEmbed uint nEmbed;: Undocumented in source. Be warned that the author may not have intended to support it.

Variables

nVocab uint nVocab;: Undocumented in source.
weight Variable!(T, 2, Storage) weight;: Undocumented in source.

Meta

Source

See Implementation