torch:embedding

(torch:embedding num-embeddings embedding-dim)

埋め込みテーブル (PyTorch の nn.Embedding) を返します。フィールドは :weight ひとつで、PyTorch のデフォルトと同じく標準正規分布から引かれた (num-embeddings embedding-dim) のパラメータです。順伝播は任意の形の整数インデックスを取り、埋め込み軸を末尾に付けて返します。同じ行が 2 回選ばれた場合は両方の勾配が蓄積されます (torch:index-select の随伴)。