torch.unique 不适用于浮点张量

Question

我正在尝试从浮点张量中提取独特元素。 我试过了：

out = torch.unique(my_tensor)

但是，此方法仅适用于 int/long 张量。我的张量是以非均匀方式量化的张量，因此它保证有一小部分浮点值。

Answer 1

您可以改用numpy.unique

import torch
import numpy as np

t = torch.tensor([1.05, 1.05, 2.01, 2.01, 3.9, 3.9001])
print(np.unique(t.numpy()))

输出：

[1.05   2.01   3.9    3.9001]