RuntimeError Traceback (most recent call last)
in <cell line: 69>()
67 vkm = VecKM()
68 pts = torch.rand((10,1000,3))
---> 69 print(vkm(pts).shape) # it will be Complex(10,1000,256)
70 pts = torch.rand((1000,3))
71 print(vkm(pts).shape) # it will be Complex(1000, 256)
2 frames
in forward(self, pts)
57 eB.transpose(-1,-2) @ eA # Real(..., 2p, 2d)
58 ) # Real(..., n, 2d)
---> 59 G = torch.complex(
60 G[:,:self.d], G[:,self.d:]
61 ) / torch.complex(
RuntimeError: The size of tensor a (256) must match the size of tensor b (1744) at non-singleton dimension 1