Then we use Vi and Vj to convert to KeOps LazyTensor objects ... e_i = Vi(torch.rand(M, D).type(tensor)) x_j = Vj(x) D2xx = LazyTensor.sum((x_i - x_j) ** 2) ...
確定! 回上一頁