paining

torch.pow() function and sum() function keep their own results so they needs too much gpu memory. (distance table is too large) When I use torch.cdist(x, y, p), it needs small...

> > torch.pow() function and sum() function keep their own results so they needs too much gpu memory. (distance table is too large) > > When I use torch.cdist(x, y,...

> > > > torch.pow() function and sum() function keep their own results so they needs too much gpu memory. (distance table is too large) > > > > When...