torch.unique does not work for float tensors

后端 未结 0 1000
南方客
南方客 2020-12-13 18:49

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

相关标签:
回答
  • 消灭零回复
提交回复
热议问题