以上就是余弦相似度的所有内容
import torch
import torch.nn.functional as F
##特征向量a
a = torch.tensor([[1, 2], [3, 4]], dtype=torch.float32)
##特征向量b
b = torch.tensor([[5, 6], [7, 8]], dtype=torch.float32)
print(F.cosine_similarity(a, b, dim=-1))
以上就是余弦相似度的所有内容
import torch
import torch.nn.functional as F
##特征向量a
a = torch.tensor([[1, 2], [3, 4]], dtype=torch.float32)
##特征向量b
b = torch.tensor([[5, 6], [7, 8]], dtype=torch.float32)
print(F.cosine_similarity(a, b, dim=-1))