pytorch中嵌入相似度的实现
torch-embed-sim的Python项目详细描述
pytorch嵌入相似性
安装
pip install torch-embed-sim
用法
fromtorch_embed_simimportEmbeddingSimclassNet(nn.Module):def__init__(self):super(Net,self).__init__()self.embed=torch.nn.Embedding(num_embeddings=10,embedding_dim=20)self.embed_sim=EmbeddingSim(num_embeddings=10)defforward(self,x):returnself.embed_sim(self.embed(x),self.embed.weight)