2020-04-18から1日間の記事一覧
いつもtorch.nn.Embeddingの意味合いを忘れるのでメモ。 import torch from torch import nn embed = nn.Embedding(num_embeddings=4, embedding_dim=10, padding_idx=0) # num_embeddingsが種類の数、embedding_dimはベクトル表現の次元数、padding_idxがpa…
いつもtorch.nn.Embeddingの意味合いを忘れるのでメモ。 import torch from torch import nn embed = nn.Embedding(num_embeddings=4, embedding_dim=10, padding_idx=0) # num_embeddingsが種類の数、embedding_dimはベクトル表現の次元数、padding_idxがpa…