Ensure hamming similarity inputs are vsa tensors (#159)

hyperdimensional-computing · Sep 26, 2023 · 2de23fd · 2de23fd
1 parent 83fdf34
commit 2de23fd
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/torchhd/functional.py b/torchhd/functional.py
@@ -1018,6 +1018,9 @@ def hamming_similarity(input: VSATensor, others: VSATensor) -> LongTensor:
                 [5, 3, 6]])
 
     """
+    input = ensure_vsa_tensor(input)
+    others = ensure_vsa_tensor(others)
+
     if input.dim() > 1 and others.dim() > 1:
         equals = input.unsqueeze(-2) == others.unsqueeze(-3)
         return torch.sum(equals, dim=-1, dtype=torch.long)