We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 51fdc33 commit ca91494Copy full SHA for ca91494
kornia/feature/integrated.py
@@ -50,7 +50,7 @@ def get_laf_descriptors(
50
timg: Tensor = img
51
if lafs.shape[1] == 0:
52
warnings.warn(f"LAF contains no keypoints {lafs.shape}, returning empty tensor")
53
- return torch.empty(lafs.shape[0], lafs.shape[1], 128)
+ return torch.empty(lafs.shape[0], lafs.shape[1], 128, dtype=lafs.dtype, device=lafs.device)
54
if grayscale_descriptor and img.size(1) == 3:
55
timg = rgb_to_grayscale(img)
56
0 commit comments