-record(embedding, {
    num_embeddings :: integer(),
    embedding_dim :: integer(),
    weight :: viva_tensor@tensor:tensor()
}).