汉明距离计算公式 pytorch

在PyTorch中，可以使用以下公式计算汉明距离：

import torch

def hamming_distance(x, y):
    # 将x和y转换为二进制字符串
    x_binary = torch.tensor(list(format(x, 'b')))
    y_binary = torch.tensor(list(format(y, 'b')))
    
    # 计算不同位数的数量
    diff = torch.sum(x_binary != y_binary)
    
    return diff.item()

这个函数接受两个整数作为输入，并返回它们的汉明距离。首先，我们将输入的整数转换为二进制字符串，然后使用逐位比较的方式计算不同位数的数量。最后，我们使用`item()`方法将结果转换为标量值。