embedding_module.hpp File Reference

Lookup-table (row-select) layer – y = W[index], batched (N, L) -> (N, L, embedding_dim). More...

#include <initializer_list>
#include <vector>
#include "pulsatrix/module.hpp"

Include dependency graph for embedding_module.hpp:

Go to the source code of this file.

Classes
class	pulsatrix::EmbeddingModule
	Embedding lookup table, rank-2 input (N, L) of float-encoded indices -> rank-3 output (N, L, embedding_dim). Structurally unlike every other module in this codebase: forward is a pure selection (row copy), with no arithmetic mixing across input features. More...

Namespaces
namespace	pulsatrix

Detailed Description

Lookup-table (row-select) layer – y = W[index], batched (N, L) -> (N, L, embedding_dim).