I am trying to extract the unique elements from a float tensor. I have tried :
out = torch.unique(my_tensor)
However this method only works f